[][src]Module tch::nn::optimizer

Structs

Adam

Parameters for the Adam optimizer.

Optimizer
RmsProp

Parameters for the RmsProp optimizer.

Sgd

Parameters for the SGD optimizer.

Functions

adam
rms_prop
sgd