a training hook for saving and loading the validated model
More...
|
def | __init__ (self, filename, model) |
| hook constructor More...
|
|
def | begin (self) |
| this will be run at session creation
|
|
def | after_create_session (self, session, _) |
| this will be run after session creation
|
|
def | save (self) |
| save the current parameters
|
|
def | restore (self) |
| restore the previously validate parameters
|
|
a training hook for saving and loading the validated model
§ __init__()
def nabu.neuralnetworks.components.hooks.ValidationSaveHook.__init__ |
( |
|
self, |
|
|
|
filename, |
|
|
|
model |
|
) |
| |
hook constructor
- Parameters
-
filename | where the model will be saved |
model | the model that will be saved |
The documentation for this class was generated from the following file: