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

More training scripts

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-06 01:38:21 +00:00
parent 733b3c5387
commit 208091ce8a
5 changed files with 43 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

3
_ShA-train.sh
View file

@ -30,5 +30,6 @@ python train.py \
--save_path ./save/ShanghaiA-stn-$(date -Iminutes) \
--batch_size 4 \
--gpus 0,1,2,3 \
--print_freq 100
--print_freq 100 \
--epochs 1000