#!/bin/bash

savepath=${1}
trainset=${2}
testset=${3}
DA=${4}
epoch=${5}
loss=${6}

CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch \
--nproc_per_node=8 --master_port 23343 train.py \
--savepath $savepath/ \
--root ~/autodl-fs/checkpoints/ \
--train_bs 1024 \
--test_bs 1000 \
--scheduler cosineanneal \
--training_epoch $epoch \
--warm_epoch 5 \
--init_lr 0.1 \
--scale_factor 8 \
--save_last_best 1 \
--tb_path /root/tf-logs \
--tb_group $savepath/ \
--trainset $trainset \
--testset $testset \
--lossfunc $loss \
--DA $DA