From c24238c877c31f4088074171f31f638293eaab7a Mon Sep 17 00:00:00 2001 From: helloway Date: Tue, 22 Mar 2022 10:21:44 +0800 Subject: [PATCH] fix the bug of cifar 10 transform and pytest --- .github/workflows/install_and_test.yml | 3 +++ tinyms/vision/configs/transform_config.yaml | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/.github/workflows/install_and_test.yml b/.github/workflows/install_and_test.yml index ffdac294..51a4ab38 100644 --- 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 index c48ce429..ef7dbcc1 100644 --- 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