fix bug (#146)

Oneflow-Inc · Feb 14, 2022 · 0506fc9 · 0506fc9
1 parent dac9569
commit 0506fc9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/projects/benchmark/classification/benchmark.py b/projects/benchmark/classification/benchmark.py
@@ -23,6 +23,7 @@ def param_count(model):
 
 
 def get_mean_std(mode="imagenet_default_mean_std"):
+    mode = mode.upper()
     if mode == "IMAGENET_DEFAULT_MEAN_STD":
         mean = (0.485, 0.456, 0.406)
         std = (0.229, 0.224, 0.225)