Définit les hyper-paramètres de l'optimiseur

🌐

Set optimizer hyper-parameters

Cette fonction est utilisée pour définir les hyper-paramètres à l'initialisation de l'optimiseur.

Utilisation

set_opt_hparams(module, ...)

Arguments

module

Un nn_module après son initialisation par setup().

...

Les paramètres passés ici seront utilisés pour initialiser les optimiseurs. Par exemple, si votre optimiseur est optim_adam et vous passez lr=0.1, alors la fonction optim_adam est appelée avec optim_adam(parameters, lr=0.1) lors de l'entraînement du modèle.

Valeur de retour

The même module luz

Voir également

Autres set_hparam: set_hparams()