osc.train.run_train_epoch

run_train_epoch(cfg, ds_train, epoch, model, optimizer, scheduler, step_counter, loss_fn_global, loss_fn_objects)[source]

Run one epoch of training.

Parameters
  • cfg (DictConfig) –

  • ds_train (Iterable) –

  • epoch (int) –

  • model (Model) –

  • optimizer (Optimizer) –

  • scheduler (_LRScheduler) –

  • step_counter (StepCounter) –

  • loss_fn_global (ModelLoss) –

  • loss_fn_objects (ModelLoss) –