Create run.sh

gabrieldernbach · Mar 12, 2024 · c26227e · c26227e
1 parent 93b9e92
commit c26227e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/run.sh b/run.sh
@@ -0,0 +1,9 @@
+python -m torch.distributed.launch \
+        --nnodes=1 --node_rank=0 \
+        --nproc_per_node=1 \
+        --master_addr="127.0.0.1" \
+        --master_port=29501 \
+        classification/main.py --cfg classification/configs/vssm1/vssm_tiny_224_0229.yaml \
+                --batch-size 128 \
+                --data-path dataset \
+                --output /tmp