discopat.nn_training.detr.engine

Train and eval functions used in main.py.

Functions

evaluate(model, criterion, postprocessors, ...)
train_one_epoch(model, criterion, ...[, ...])

discopat.nn_training.detr.engine.evaluate(model, criterion, postprocessors, data_loader, base_ds, device, output_dir)

discopat.nn_training.detr.engine.train_one_epoch(model, criterion, data_loader, optimizer, device, epoch, max_norm=0)

Parameters:

• model (Module)

• criterion (Module)

• data_loader (Iterable)

• optimizer (Optimizer)

• device (device)

• epoch (int)

• max_norm (float)