discopat.nn_training.torch

Classes

TorchNNTrainer(net, dataset, val_dataset, ...)

class discopat.nn_training.torch.TorchNNTrainer(net, dataset, val_dataset, parameters, device, callbacks=None)

Bases: NNTrainer

Parameters:

• net (NeuralNet)

• dataset (Dataset)

• val_dataset (Dataset)

• parameters (dict[str, Any])

• device (ComputingDevice)

• callbacks (list or None)

set_default_lr_scheduler()

Return type:

LRScheduler

set_default_optimiser()

Return type:

Optimizer

train(num_epochs)

Parameters:

num_epochs (int)