Contents:
Run one epoch of training.
cfg (DictConfig) –
DictConfig
ds_train (Iterable) –
Iterable
epoch (int) –
int
model (Model) –
Model
optimizer (Optimizer) –
Optimizer
scheduler (_LRScheduler) –
_LRScheduler
step_counter (StepCounter) –
StepCounter
loss_fn_global (ModelLoss) –
ModelLoss
loss_fn_objects (ModelLoss) –