torchgpe.utils.callbacks.Callback.on_epoch_end

Callback.on_epoch_end(epoch)

Function called by the gpe.bec2D.gas.Gas at the end of each epoch

Parameters:

epoch (int) – The epoch number