Skip to content

NeuralNetworks.Trainer

class NeuralNetworks.Trainer (*nets, inputs, outputs, init_train_size, final_train_size, optim, init_lr, final_lr, crit, batch_size) [source]

Classe pour entraîner des réseaux avec mini-batchs et Automatic Mixed Precision.

Paramètres Type Optionnel
*nets Module Non
inputs torch.Tensor([float]) Non
outputs torch.Tensor([float]) Non
init_train_size float Oui
final_train_size float Oui
optim optim Oui
init_lr float Oui
final_lr float Oui
crit crit Oui
batch_size int Oui
Initialisation d'un trainer
Trainer_exemple.py
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
from NeuralNetworks import Trainer

T = Trainer (
    net                       , # (1)!
    inputs           = inputs , # (2)!
    outputs          = outputs, # (3)!
    init_train_size  = 0.001  , 
    final_train_size = 1      , 
    init_lr          = 1e-3   , 
    final_lr         = 1e-5   , 
    optim            = 'Adam' , 
    crit             = 'MSE'  , 
    batch_size       = 163840   
)
  1. Voir Module
  2. La définition de inputs n'est pas explicitée ici
  3. La définition de outputs n'est pas explicitée ici