update runtime.txt

Cathy0908 · Aug 16, 2023 · 9aa11c7 · 9aa11c7
1 parent 1dc5e2f
commit 9aa11c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/configs/classification/imagenet/common/dataset/imagenet_classification.py b/configs/classification/imagenet/common/dataset/imagenet_classification.py
@@ -153,7 +153,7 @@
         data=data['val'],
         dist_eval=True,
         evaluators=[
-            dict(type='ClsEvaluator', topk=(1, 5), class_list=class_list)
+            dict(type='ClsEvaluator', topk=(1, ), class_list=class_list)
         ],
     )
 ]
diff --git a/requirements/runtime.txt b/requirements/runtime.txt
@@ -25,6 +25,7 @@ rapidfuzz
 scikit-image
 scikit-learn
 seaborn
+shapely==1.8.4
 tensorboard
 thop
 timm==0.5.4