9 lines
314 B
Plaintext
9 lines
314 B
Plaintext
# This installs Pytorch for CUDA 8 only. If you are using a newer version,
|
|
# please visit http://pytorch.org/ and install the relevant version.
|
|
torch>=0.4.1,<0.5.0
|
|
# progress bars in model download and training scripts
|
|
tqdm
|
|
# Accessing files from S3 directly.
|
|
boto3
|
|
# Used for downloading models over HTTP
|
|
requests |