tabensemb.model.AbstractNN.on_train_batch_start

method

AbstractNN.on_train_batch_start(batch: Any, batch_idx: int) → int | None

Called in the training loop before anything happens for that batch.

If you return -1 here, you will skip training for the rest of the current epoch.

Parameters:

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

• batch_idx – the index of the batch