save
open override fun save(modelDirectory: File, savingFormat: SavingFormat, saveOptimizerState: Boolean, writingMode: WritingMode)
Content copied to clipboard
Saves the model as graph and weights.
Parameters
modelDirectory
Path to model directory.
savingFormat
One of approaches to store model configurations and weights.
saveOptimizerState
Saves internal optimizer states (variables) if true.
writingMode
Default behaviour of handling different edge cases with existing directory before model saving.
Throws
FileNotFoundException
If modelDirectory does not contain all required files.