Fix

Ailurus1 · Nov 25, 2022 · 6b3aef6 · 6b3aef6
1 parent 9c87513
commit 6b3aef6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py b/cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py
@@ -1,7 +1,7 @@
 import os
 import sys
 
-if os.environ.get('CM_IMAGENET_QUANTIZED', False):
+if os.environ.get('CM_IMAGENET_QUANTIZED', False) == "yes":
     import generic_preprocess
     generic_preprocess.preprocess()
 else: