tabensemb.model.AbstractNN.on_predict_batch_start

method

AbstractNN.on_predict_batch_start(batch: Any, batch_idx: int, dataloader_idx: int) → None

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

Parameters:

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

• batch_idx – the index of the batch

• dataloader_idx – the index of the dataloader