run_pretraining.sh

#!/usr/bin/env bash

data_dir="data"
if [ ! -x "$data_dir" ]; then
  ln -s /data/swin/train/ $data_dir
fi

export HIP_VISIBLE_DEVICES=0,1,2,3
torchrun --nproc_per_node=4 train_cifar10.py --net swin --n_epochs 500 --noaug $1 $2