Checkpoint - Githubissues

fyu / drn

Dilated Residual Networks

https://www.vis.xyz/pub/drn

BSD 3-Clause "New" or "Revised" License

1.1k stars 219 forks source link

Closed smitheric95 closed 4 years ago

smitheric95 commented 4 years ago

This PR adds some simple functionality for a cleaner working directory:

Adds the save_path argument to specify where model weight checkpoints are saved.
Adds the save_iter argument to specify how many training iterations should pass between checkpoint saves.

Thanks for the great paper/code! :smiley:

fyu commented 4 years ago

Thanks!