Train Engine

biapy.engine.train_engine.train_one_epoch(cfg, model, model_call_func, loss_function, activations, metric_function, prepare_targets, data_loader, optimizer, device, loss_scaler, epoch, log_writer=None, lr_scheduler=None, start_steps=0, verbose=False)[source]
biapy.engine.train_engine.evaluate(cfg, model, model_call_func, loss_function, activations, metric_function, prepare_targets, epoch, data_loader, lr_scheduler)[source]