Update run.sh

monster119120 · Apr 8, 2023 · b8b48fd · b8b48fd
1 parent 60f6753
commit b8b48fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/run.sh b/run.sh
@@ -1 +1,2 @@
-CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python main.py --arch resnet50_dynamic --gpus 8 --lr 0.001 --data-path ../data/imagenet --pretrained --strategy ddp_find_unused_parameters_false
+CUDA_VISIBLE_DEVICES=7 python main.py --arch resnet50 --pretrained --data-path ../data/imagenet -e \
+    --accelerator gpu --devices 1