class TerminateOnTrainNaN[source]

TerminateOnTrainNaN(after_create=None, before_fit=None, before_epoch=None, before_train=None, before_batch=None, after_pred=None, after_loss=None, before_backward=None, before_step=None, after_cancel_step=None, after_step=None, after_cancel_batch=None, after_batch=None, after_cancel_train=None, after_train=None, before_validate=None, after_cancel_validate=None, after_validate=None, after_cancel_epoch=None, after_epoch=None, after_cancel_fit=None, after_fit=None) :: Callback

A Callback that terminates training if the training loss is NaN and ignores valid loss.

class SaveModelAtEnd[source]

SaveModelAtEnd(fname='model', with_opt=False) :: SaveModelCallback

A SaveModelCallback which only saves the model at the end so loggers can find it.

	Type	Default	Details
fname	str	model	Model filename
with_opt	bool	False	Include optimizer state