m3gnet.callbacks module

Callback functions

class m3gnet.callbacks.ManualStop

Bases: keras.callbacks.Callback

Stop the training manually by putting a “STOP” file in the directory

on_batch_end(epoch: int, logs: Optional[Dict] = None) → None

Codes called at the end of a batch Args:

epoch (int): epoch id logs (Dict): log dict

Returns: None