osc.train.run_train_val_viz_epochs

run_train_val_viz_epochs(cfg, model, optimizer, scheduler, ds_train, ds_val, ds_test, ds_vqa, viz_batch, loss_fn_global, loss_fn_objects)

Run train, val, viz for a certain number of epochs. Handle CTRL+C gracefully.

Parameters

• cfg (DictConfig) –

• model (Model) –

• optimizer (Optimizer) –

• scheduler (_LRScheduler) –

• ds_train (Iterable) –

• ds_val (Iterable) –

• ds_test (DatasetV2) –

• ds_vqa (DatasetV2) –

• viz_batch (ndarray) –

• loss_fn_global (ModelLoss) –

• loss_fn_objects (ModelLoss) –