fix moco script

vturrisi · Jun 9, 2022 · 0b9c9f3 · 0b9c9f3
1 parent f3c45ba
commit 0b9c9f3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bash_files/pretrain/imagenet/mocov2plus.sh b/bash_files/pretrain/imagenet/mocov2plus.sh
@@ -6,7 +6,8 @@ python3 main_pretrain.py \
     --val_dir imagenet/val \
     --max_epochs 100 \
     --devices 0,1 \
-    --accelerator ddp \
+    --accelerator gpu \
+    --strategy ddp \
     --sync_batchnorm \
     --precision 16 \
     --optimizer sgd \