set_hparams: Set hyper-parameter of a module

In mlverse/torchlight: Higher Level 'API' for 'torch'

Set hyper-parameter of a module

Description

This function is used to define hyper-parameters before calling fit for luz_modules.

Usage

set_hparams(module, ...)

Arguments

module	An nn_module that has been setup().
...	The parameters set here will be used to initialize the nn_module, ie they are passed unchanged to the initialize method of the base nn_module.

Value

The same luz module

See Also

Other set_hparam: set_opt_hparams()

mlverse/torchlight documentation built on Sept. 19, 2024, 11:22 p.m.