discopat.nn_training.detr.detr_trainer

Classes

DetrTrainer(net, dataset, val_dataset, ...)

class discopat.nn_training.detr.detr_trainer.DetrTrainer(net, dataset, val_dataset, parameters, device, callbacks=None, num_classes=0)[source]

Bases: NNTrainer

Parameters:

net (NeuralNet)
dataset (Dataset)
val_dataset (Dataset)
parameters (dict[str, Any])
device (ComputingDevice)
callbacks (list)

set_default_lr_scheduler()[source]

Return type:: LRScheduler

set_default_optimiser()[source]

Return type:: Optimizer

train(num_epochs)[source]

Parameters:: num_epochs (int)