tabensemb.model.AbstractNN.on_train_epoch_end

method

AbstractNN.on_train_epoch_end() → None

Called in the training loop at the very end of the epoch.

To access all batch outputs at the end of the epoch, either:

1. Implement training_epoch_end in the LightningModule OR

2. Cache data across steps on the attribute(s) of the LightningModule and access them in this hook