netket.callbacks.Timeout#

class netket.callbacks.Timeout[source]#

Bases: object

A simple callback to stop NetKet after some time has passed.

This callback monitors whether a driver has been training for more than a given timeout in order to hard stop training.

Attributes

timeout: float#: Number of seconds to wait before the training will be stopped.

Methods

__call__(step, log_data, driver)[source]#

A boolean function that determines whether or not to stop training.

Parameters:

step – An integer corresponding to the step (iteration or epoch) in training.
log_data – A dictionary containing log data for training.
driver – A NetKet variational driver.

Returns:

A boolean. If True, training continues, else, it does not.

Note

This callback does not make use of step, log_data nor driver.

replace(**updates)#: Returns a new object replacing the specified fields with new values.

netket.callbacks.Timeout