fix the bug of cifar 10 transform and pytest

tinyms-ai · Mar 22, 2022 · c24238c · c24238c
1 parent 4a1b90b
commit c24238c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.github/workflows/install_and_test.yml b/.github/workflows/install_and_test.yml
@@ -5,6 +5,9 @@ on:
   pull_request:
     branches: [ main ]
 
+env:
+  MS_DEV_ENABLE_FALLBACK: 0
+
 jobs:
   pytest:
     strategy:

diff --git a/tinyms/vision/configs/transform_config.yaml b/tinyms/vision/configs/transform_config.yaml
@@ -21,7 +21,7 @@ Cifar10Transform:
   labels: ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
   transform_strategy: ['TOP1_CLASS', 'TOP3_CLASS', 'TOP5_CLASS']
   random_crop:
-    size: [224, 224]
+    size: [32, 32]
     padding: [4, 4, 4, 4]
   random_horizontal_flip:
     prob: 0.5