diff --git a/demo/autofinetune/run_autofinetune.sh b/demo/autofinetune/run_autofinetune.sh index 7e0eead1bf7861d315186b80eb69c1480c6611dd..1db9750604fdda49c8afb1a1d215e44326dc25d7 100644 --- a/demo/autofinetune/run_autofinetune.sh +++ b/demo/autofinetune/run_autofinetune.sh @@ -2,7 +2,7 @@ OUTPUT=result/ hub autofinetune img_cls.py \ --param_file=hparam.yaml \ - --cuda=['6'] \ + --cuda=['0'] \ --popsize=5 \ --round=10 \ --output_dir=${OUTPUT} \