zc/mlp-project
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Sync

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-03 22:19:47 +00:00
parent fc941ebaf7
commit c4905acf6d
2 changed files with 8 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

3
_ShanghaiA-train.sh
View file

@ -22,8 +22,7 @@ export TMP=/disk/scratch/${STUDENT_ID}/
source /home/${STUDENT_ID}/miniconda3/bin/activate mlp-cuda
python train.py \
--debug=True \
--model='stn' \
--save_path ./save_file/ShanghaiA \
--batch_size 4 \
# --gpus 0,1,2,3,4,5 \
--gpus 0,1,2,3,4,5