File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 3535
3636```
3737 --epochs INT Number of epochs. Default is 100.
38- --batch-size INT Batch size. Default is 256 .
38+ --batch-size INT Batch size. Default is 64 .
3939 --learning-rate FLOAT Adam learning rate. Default is 0.001.
4040 --decay-rate FLOAT Rate of decay for learning rate. Default is 0.95.
4141 --decay-steps INT How many steps before decy lr. Default is 500.
42- --evaluate-steps INT How many steps to evluate val set. Default is 500 .
42+ --evaluate-steps INT How many steps to evluate val set. Default is 50 .
4343 --l2-lambda FLOAT L2 regularization lambda. Default is 0.0.
44- --checkpoint-steps INT How many steps to save model. Default is 500 .
44+ --checkpoint-steps INT How many steps to save model. Default is 50 .
4545 --num-checkpoints INT Number of checkpoints to store. Default is 10.
4646```
4747
You can’t perform that action at this time.
0 commit comments