Skip to content

Command line arguments for learning rates and RMSprop vs Adam

Jakob Schiøtz requested to merge schiotz/NeuralNetwork_HRTEM:learningrates into master

  • Command line argument for switching to RMSprop and setting learning rate.

  • Command line argument for specifying parameters to the Adam optimizer.

  • Change default optimizer to Adam(amsgrad=True).

  • Per default, only checkpoint every fifth epoch to save disk space. Override on command line.

Edit:

  • BUG FIX: Correct network parameters.
Edited by Jakob Schiøtz

Merge request reports