glenn-jocher
commited on
Commit
•
b6fe2e4
1
Parent(s):
07493a7
Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -10,7 +10,7 @@ PyYAML>=5.3
|
|
10 |
torchvision>=0.6
|
11 |
scipy
|
12 |
tqdm
|
13 |
-
|
14 |
|
15 |
# Nvidia Apex (optional) for mixed precision training --------------------------
|
16 |
# 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
|
|
|
10 |
torchvision>=0.6
|
11 |
scipy
|
12 |
tqdm
|
13 |
+
pycocotools>=2.0
|
14 |
|
15 |
# Nvidia Apex (optional) for mixed precision training --------------------------
|
16 |
# 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
|