Glenn Jocher
GitHub
4 years ago
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with
1 additions and
1 deletions
-
requirements.txt
|
|
@@ -10,7 +10,7 @@ PyYAML>=5.3 |
|
|
|
torchvision>=0.6 |
|
|
|
scipy |
|
|
|
tqdm |
|
|
|
git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI |
|
|
|
pycocotools>=2.0 |
|
|
|
|
|
|
|
# Nvidia Apex (optional) for mixed precision training -------------------------- |
|
|
|
# git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex |