shen/deeplearning-models
1
0
Fork 0
mirror of https://github.com/rasbt/deeplearning-models.git synced 2024-11-25 19:12:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix some args

Browse Source
This commit is contained in:
rasbt 2022-02-27 20:18:04 -06:00
parent 1f4b2ed40a
commit 56da2dcf6d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
templates/pytorch_lightning/submit_command.sh
View File

@ -3,6 +3,6 @@ grid run \
--framework lightning \
--gpus 2 \
tune_classification_basic.py \
--learning_rate "[0.0001, 0.0005, 0.001, 0.005, 0.01, 0.05)" \
--batch_size "[64, 128, 256]"\
--learning_rate "[0.0001, 0.0005, 0.001, 0.005, 0.01, 0.05]" \
--batch_size "[64, 128, 256]" \
--epochs 20