Merge branch 'main' of

github.com:Oneflow-Inc/vision into add-SEModule-doc add SEModule docstring
Oneflow-Inc · Feb 16, 2022 · 272f559 · 272f559
2 parents 1eb58de + 0506fc9
commit 272f559
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
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)

diff --git a/setup.py b/setup.py
@@ -25,7 +25,7 @@ def get_dist(pkgname):
  return None
 
 
-version = "0.0.6"
+version = "0.1.0"
 package_name = "flowvision"
 cwd = os.path.dirname(os.path.abspath(__file__))