diff --git a/demo/auto-compression/run_imagenet.sh b/demo/auto-compression/run_imagenet.sh index 3ddcf07f2cd7d5bf67df9d6f25c8c49f97a3fe3a..376554f37e5dbfa9e356862ec6e0fb93fa460b57 100644 --- a/demo/auto-compression/run_imagenet.sh +++ b/demo/auto-compression/run_imagenet.sh @@ -5,5 +5,5 @@ python3.7 demo_imagenet.py \ --save_dir='./save_qat_mbv2/' \ --devices='cpu' \ --batch_size=2 \ - --data_dir='data/ILSVRC2012/' \ + --data_dir='../data/ILSVRC2012/' \ --config_path='./configs/CV/mbv2_ptq_hpo.yaml'