fix efficientnet imagenet accuracy

2021-07-09 05:08:54 +08:00 · 2021-07-09 05:08:54 +08:00 · 61d1559bb8
parent 7dd2ae02ee
commit 61d1559bb8
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/cv/efficientnet/src/config.py
+++ b/model_zoo/official/cv/efficientnet/src/config.py
@ -68,7 +68,7 @@ efficientnet_b0_config_imagenet = EasyDict({
    'workers': 8,
    'amp_level': 'O0',
    'opt': 'rmsprop',
-    'num_classes': 10,
+    'num_classes': 1000,
    #'Type of global pool, "avg", "max", "avgmax", "avgmaxc"
    'gp': 'avg',
    'momentum': 0.9,