netket.callbacks.Timeout

class netket.callbacks.Timeout(timeout)[source]

Bases: object

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

__init__(timeout)[source]

Constructs a new Timeout object that monitors whether a driver has been training for more than a given timeout in order to hard stop training.

Parameters

timeout – Number of seconds to wait before hard stopping training.

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.

reset()[source]

Resets the initial time of the training