Skip to content

Commit

Permalink
Merge pull request #1071 from Belerafon/master
Browse files Browse the repository at this point in the history
Doc update according ability to reset optimizer
  • Loading branch information
vince62s committed Nov 25, 2018
2 parents 685be43 + 4accba7 commit b085d57
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions docs/source/options/train.md
Expand Up @@ -145,6 +145,13 @@ number of steps
* **-save_checkpoint_steps [5000]**
Save a checkpoint every X steps

* **-reset_optim [none]**
Ability to reset optimizer. Options:
"all": reset completely the optimizer (train_steps, type of optim, ....),
"states": load everything from the checkpoint except Adam states,
"keep_states": load Adam states from the checkpoint but apply
command line changes

* **-keep_checkpoint [-1]**
Keep X checkpoints (negative: keep all)

Expand Down

0 comments on commit b085d57

Please sign in to comment.