discopat.nn_training.nn_trainer

Classes

NNTrainer(net, dataset, val_dataset, ...[, ...])

class discopat.nn_training.nn_trainer.NNTrainer(net, dataset, val_dataset, parameters, device, callbacks=None)[source]

Bases: ABC

Parameters:

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

static parse_parameters(parameters)[source]

Parameters:: parameters (dict[str, Any])
Return type:: tuple(dict, dict, dict)

abstractmethod set_default_lr_scheduler()[source]

Return type:: Any

abstractmethod set_default_optimiser()[source]

Return type:: Any

abstractmethod train()[source]

Return type:: None