neuralop.training.callbacks.BasicLoggerCallback

class neuralop.training.callbacks.BasicLoggerCallback(wandb_kwargs=None)[source]

Callback that implements simple logging functionality expected when passing verbose to a Trainer

Methods

on_batch_start
on_before_loss
on_before_val
on_epoch_start
on_init_end
on_train_start
on_val_end
on_val_epoch_end