Caffe2 - Python API
A deep learning, cross platform ML framework
Functions | Variables
net_construct_bench Namespace Reference

Module caffe2.experiments.python.net_construct_bench. More...

Functions

def AddMomentumParameterUpdate (train_model, LR)
 
def Create (args)
 
def main ()
 

Variables

 log = logging.getLogger("net_construct_bench")
 
 sort
 

Detailed Description

Module caffe2.experiments.python.net_construct_bench.

Function Documentation

def net_construct_bench.AddMomentumParameterUpdate (   train_model,
  LR 
)
Add the momentum-SGD update.

Definition at line 28 of file net_construct_bench.py.