set_opt_hparams

This function is used to define hyper-parameters for the optimizer initialization
method.

A high level interface for 'torch' providing utilities to reduce the
the amount of code needed for common tasks, abstract away torch details and
make the same code work on both the 'CPU' and 'GPU'. It's flexible enough to
support expressing a large range of models. It's heavily inspired by 'fastai' by
Howard et al. (2020) <doi:10.48550/arXiv.2002.04688>, 'Keras' by Chollet et al. (2015) and
'PyTorch Lightning' by Falcon et al. (2019) <doi:10.5281/zenodo.3828935>.

Daniel Falbel

Higher Level 'API' for 'torch'

Christophe Regouby

 RStudio

set_opt_hparams function

<dl><dt>module</dt>
<dd>An <code>nn_module</code> that has been <code>setup()</code>.</dd>
<dt>...</dt>
<dd>The parameters passed here will be used to initialize the optimizers.
For example, if your optimizer is <code>optim_adam</code> and you pass <code>lr=0.1</code>, then the
<code>optim_adam</code> function is called with <code>optim_adam(parameters, lr=0.1)</code> when fitting
the model.</dd></dl>

Arguments

Set optimizer hyper-parameters — set_opt_hparams

<dl>

<dt>module</dt>
<dd>An <code>nn_module</code> that has been <code>setup()</code>.</dd>


<dt>...</dt>
<dd>The parameters passed here will be used to initialize the optimizers.
For example, if your optimizer is <code>optim_adam</code> and you pass <code>lr=0.1</code>, then the
<code>optim_adam</code> function is called with <code>optim_adam(parameters, lr=0.1)</code> when fitting
the model.</dd>

</dl>

set_opt_hparams: Set optimizer hyper-parameters

Description

Usage

Value

Arguments

See Also