6 lines
163 B
Bash
6 lines
163 B
Bash
FOLDER_NAME='tr_scorenet_nb201'
|
|
|
|
CUDA_VISIBLE_DEVICES=$1 python main.py --config configs/tr_scorenet.py \
|
|
--mode train \
|
|
--config.folder_name $FOLDER_NAME
|