Package | Description |
---|---|
org.deeplearning4j.earlystopping | |
org.deeplearning4j.earlystopping.saver |
Modifier and Type | Method and Description |
---|---|
EarlyStoppingConfiguration.Builder<T> |
EarlyStoppingConfiguration.Builder.modelSaver(EarlyStoppingModelSaver<T> modelSaver)
How should models be saved?
|
Modifier and Type | Class and Description |
---|---|
class |
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
|
class |
LocalFileGraphSaver
Save the best (and latest/most recent)
ComputationGraph s learned during early stopping training to the local file system. |
class |
LocalFileModelSaver
Save the best (and latest/most recent) models learned during early stopping training to the local file system.
|
Copyright © 2020. All rights reserved.