Class | Description |
---|---|
InMemoryModelSaver<T extends Model> |
Save the best (and latest) models for early stopping training to memory for later retrieval
Note: Assumes that network is cloneable via .clone() method
|
LocalFileGraphSaver |
Save the best (and latest/most recent)
ComputationGraph s learned during early stopping training to the local file system.Instances of this class will save 3 files for best (and optionally, latest) models: (a) The network configuration: bestGraphConf.json (b) The network parameters: bestGraphParams.bin (c) The network updater: bestGraphUpdater.bin NOTE: The model updater is an object that contains the internal state for training features such as AdaGrad, Momentum and RMSProp. The updater is not required to use the network at test time; it is saved in case further training is required. |
LocalFileModelSaver |
Save the best (and latest/most recent) models learned during early stopping training to the local file system.
Instances of this class will save 3 files for best (and optionally, latest) models: (a) The network configuration: bestModelConf.json (b) The network parameters: bestModelParams.bin (c) The network updater: bestModelUpdater.bin NOTE: The model updater is an object that contains the internal state for training features such as AdaGrad, Momentum and RMSProp. The updater is not required to use the network at test time; it is saved in case further training is required. |