Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
torch::optim::SGDOptions Struct Referenceabstract

Public Member Functions

 SGDOptions (double learning_rate)
 
 TORCH_ARG (double, learning_rate)
 
 TORCH_ARG (double, momentum)=0
 
 TORCH_ARG (double, dampening)=0
 
 TORCH_ARG (double, weight_decay)=0
 
 TORCH_ARG (bool, nesterov)
 

Detailed Description

Definition at line 22 of file sgd.h.


The documentation for this struct was generated from the following files: