Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
torch::nn::DropoutOptions Struct Reference

Options for Dropout and FeatureDropout. More...

#include <dropout.h>

Public Member Functions

 DropoutOptions (double rate=0.5)
 
 TORCH_ARG (double, rate)
 The probability with which a particular component of the input is set to zero. More...
 

Detailed Description

Options for Dropout and FeatureDropout.

Definition at line 14 of file dropout.h.

Member Function Documentation

torch::nn::DropoutOptions::TORCH_ARG ( double  ,
rate   
)

The probability with which a particular component of the input is set to zero.

Changes to this parameter at runtime are effective.


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