Passer au contenu

Cette fonction est utilisée pour définir les hyper-paramètres avant d'appeler fit pour les luz_modules.

Utilisation

set_hparams(module, ...)

Arguments

module

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

...

Les paramètres définis ici seront utilisés pour initialiser le nn_module, c'est-à-dire qu'ils seront passés tels quels à la méthode initialize du nn_module de base.

Valeur de retour

Le même module luz

Voir également

Autres set_hparam: set_opt_hparams()