static CheckpointListener.Builder |
CheckpointListener.builder(@NonNull 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.
|