discopat.nn_training.torch_detection_utils.engine

Functions

train_one_epoch(model, optimizer, ...[, scaler])

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