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