discopat.nn_training.torch_detection_utils.engine

Functions

`evaluate`(model, data_loader, device)
`train_one_epoch`(model, optimizer, ...[, scaler])

discopat.nn_training.torch_detection_utils.engine.evaluate(model, data_loader, device)[source]

discopat.nn_training.torch_detection_utils.engine.train_one_epoch(model, optimizer, data_loader, device, epoch, print_freq, scaler=None)[source]