Package | Description |
---|---|
org.nd4j.autodiff.listeners.checkpoint |
Modifier and Type | Method and Description |
---|---|
static CheckpointListener.Builder |
CheckpointListener.builder(File rootDir) |
CheckpointListener.Builder |
CheckpointListener.Builder.deleteExisting(boolean deleteExisting)
If the checkpoint listener is set to save to a non-empty directory, should the CheckpointListener-related
content be deleted?
This is disabled by default (and instead, an exception will be thrown if existing data is found) WARNING: Be careful when enabling this, as it deletes all saved checkpoint models in the specified directory! |
CheckpointListener.Builder |
CheckpointListener.Builder.fileNamePrefix(String fileNamePrefix) |
CheckpointListener.Builder |
CheckpointListener.Builder.keepAll()
Keep all model checkpoints - i.e., don't delete any.
|
CheckpointListener.Builder |
CheckpointListener.Builder.keepLast(int n)
Keep only the last N most recent model checkpoint files.
|
CheckpointListener.Builder |
CheckpointListener.Builder.keepLastAndEvery(int nLast,
int everyN)
Keep the last N most recent model checkpoint files, and every M checkpoint files.
For example: suppose you save every 100 iterations, for 2050 iteration, and use keepLastAndEvery(3,5). |
CheckpointListener.Builder |
CheckpointListener.Builder.logSaving(boolean logSaving)
If true (the default) log a message every time a model is saved
|
CheckpointListener.Builder |
CheckpointListener.Builder.saveEvery(long amount,
TimeUnit timeUnit)
Save a model periodically
|
CheckpointListener.Builder |
CheckpointListener.Builder.saveEvery(long amount,
TimeUnit timeUnit,
boolean sinceLast)
Save a model periodically (if sinceLast == false), or if the specified amount of time has elapsed since
the last model was saved (if sinceLast == true)
|
CheckpointListener.Builder |
CheckpointListener.Builder.saveEveryEpoch()
Save a model at the end of every epoch
|
CheckpointListener.Builder |
CheckpointListener.Builder.saveEveryNEpochs(int n)
Save a model at the end of every N epochs
|
CheckpointListener.Builder |
CheckpointListener.Builder.saveEveryNIterations(int n)
Save a model every N iterations
|
CheckpointListener.Builder |
CheckpointListener.Builder.saveEveryNIterations(int n,
boolean sinceLast)
Save a model every N iterations (if sinceLast == false), or if N iterations have passed since
the last model vas saved (if sinceLast == true)
|
CheckpointListener.Builder |
CheckpointListener.Builder.saveUpdaterState(boolean saveUpdaterState)
Whether the updater state (history/state for Adam, Nesterov momentum, etc) should be saved with each checkpoint.
Updater state is saved by default. |
Copyright © 2019. All rights reserved.