11 lines
208 B
Bash
11 lines
208 B
Bash
|
FOLDER_NAME='transfer_nag_nb201'
|
||
|
|
|
||
|
|
GPU=$1
|
||
|
|
DATANAME=$2
|
||
|
|
|
||
|
|
CUDA_VISIBLE_DEVICES=$GPU python main_exp/transfer_nag/main.py \
|
||
|
|
--gpu $GPU \
|
||
|
|
--test \
|
||
|
|
--folder_name $FOLDER_NAME \
|
||
|
|
--data-name $DATANAME
|