train.sh 198 Bytes
Newer Older
Sugon_ldc's avatar
Sugon_ldc committed
1
2
3
4
#!/usr/bin/env bash
export HIP_VISIBLE_DEVICES=0,1,2,3
torchrun --nproc_per_node=4 train_cifar10.py --net swin --n_epochs 500 --noaug --log_dir pid.txt 2>&1 | tee swin_dcu_`date +%Y%m%d%H%M%S`.log