Initial release of SwiftFormer

2023-03-26 23:31:59 +04:00
commit 574907f49b
19 changed files with 1751 additions and 0 deletions
--- a/dist_test.sh
+++ b/dist_test.sh
@@ -0,0 +1,11 @@
+#!/usr/bin/env bash
+
+IMAGENET_PATH=$1
+MODEL=$2
+CHECKPOINT=$3
+nGPUs=$4
+
+python -m torch.distributed.launch --master_addr="127.0.0.1" --master_port=1234 --nproc_per_node=$nGPUs --use_env main.py --model "$MODEL" \
+--resume $CHECKPOINT --eval \
+--data-path "$IMAGENET_PATH" \
+--output_dir SwiftFormer_test