|
ocssw
V2022
|
ModelCheckpoint
Inheritance diagram for ModelCheckpoint:
Collaboration diagram for ModelCheckpoint:
Public Member Functions | |
| def | __init__ (self, path) |
| def | on_train_end (self, *args, **kwargs) |
Public Attributes | |
| tmp_folder | |
| checkpoint | |
Detailed Description
Save models during training, and load the best performing
on the validation set once training is completed.
Currently untested.
Definition at line 91 of file callbacks.py.
Constructor & Destructor Documentation
◆ __init__()
| def __init__ | ( | self, | |
| path | |||
| ) |
Definition at line 97 of file callbacks.py.
Member Function Documentation
◆ on_train_end()
| def on_train_end | ( | self, | |
| * | args, | ||
| ** | kwargs | ||
| ) |
Definition at line 105 of file callbacks.py.
Member Data Documentation
◆ checkpoint
| checkpoint |
Definition at line 100 of file callbacks.py.
◆ tmp_folder
| tmp_folder |
Definition at line 99 of file callbacks.py.
The documentation for this class was generated from the following file:
- /gfs-oceanweb/web/ocssw/ocssw_src/src/aquaverse/MDN/model/callbacks.py
