Bases: TrainTaskBegin
After predicting on a training batch, but before training on it.
-
__init__(
- train_task: int,
- global_step: int,
- batch: int,
- x: Tensor,
- y: Tensor,
- y_hat: Tensor,
) → None
-
batch: int
Batch ID within the current task stream.
-
global_step: int
A monotonically increasing integer that counts training and evaluation steps.
-
train_task: int
The ID of the training task that has begun.
-
x: Tensor
The input batch.
-
y: Tensor
The target batch.
-
y_hat: Tensor
The predicted batch.
