tabensemb.model.AbstractNN.on_train_batch_end

method

AbstractNN.on_train_batch_end(outputs: Tensor | Dict[str, Any], batch: Any, batch_idx: int) → None

Called in the training loop after the batch.

Parameters:

• outputs – The outputs of training_step_end(training_step(x))

• batch – The batched data as it is returned by the training DataLoader.

• batch_idx – the index of the batch