osc.train.run_viz

run_viz(cfg, viz_batch, epoch, model, step_counter, loss_fn_global, loss_fn_objects)

Run inference on a single batch of images and visualize everything!

Parameters

• cfg (DictConfig) –

• viz_batch (ndarray) –

• epoch (int) –

• model (Model) –

• step_counter (StepCounter) –

• loss_fn_global (ModelLoss) –

• loss_fn_objects (ModelLoss) –