netket.callbacks.InvalidLossStopping

class netket.callbacks.InvalidLossStopping

Bases: AbstractCallback

A simple callback to stop the optimisation when the monitored quantity becomes invalid for at least patience steps.

Inheritance

__init__(monitor='mean', patience=0)

Construct a callback stopping the optimisation when the monitored quantity becomes invalid for at least patience steps.

Parameters:

• monitor (str) – a string with the name of the quantity to be monitored. This is applied to the standard loss optimised by a driver, such as the Energy for the VMC driver. Should be one of ‘mean’, ‘variance’, ‘error_of_mean’ (default: ‘mean’).

• patience (int | float) – Number of steps to wait before stopping the execution after the tracked quantity becomes invalid (default 0, meaning that it stops immediately).

Attributes

callback_order

monitor: str

Loss statistic to monitor. Should be one of ‘mean’, ‘variance’, ‘error_of_mean’.

patience: int | float

Number of epochs with invalid loss after which training will be stopped.

Methods

before_parameter_update(step, log_data, driver)

Called after all update logic has been computed and the step has been accepted, but before the driver applies the parameter update.

At this point:

• The loss and its gradient have been computed by compute_loss_and_update().

• The step has been accepted (not rejected by on_compute_update_end()).

• driver.step_count still refers to the current step — it has not yet been incremented.

• The variational state parameters have not yet changed.

This is the right place to estimate additional observables, add data to log_data, or take a snapshot of the state for logging. Callbacks with a lower callback_order run first, so observables callbacks (order 0) are guaranteed to populate log_data before logger callbacks (order 10) read it.

on_compute_update_end(step, log_data, driver)

Callback called at the end of the compute update phase, after computing the loss and its gradient.

This is called before the parameters are updated, so it can be used to implement custom logic for rejecting a step based on the computed loss or gradient.

Return type:

bool

Returns:

A boolean indicating whether to reject the step (i.e. repeat it with the same parameters). If it returns None, it is treated as False.

on_compute_update_start(step, log_data, driver)

on_run_end(step, driver)

on_run_error(step, error, driver)

on_run_start(step, driver)

on_step_end(step, log_data, driver)

on_step_start(step, log_data, driver)

replace(**kwargs)

Replace the values of the fields of the object with the values of the keyword arguments. If the object is a dataclass, dataclasses.replace will be used. Otherwise, a new object will be created with the same type as the original object.

Return type:

TypeVar(P, bound= Pytree)

Parameters:

• self (P)

• kwargs (Any)