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 | |
Module caffe2.experiments.python.net_construct_bench.
def net_construct_bench.AddMomentumParameterUpdate | ( | train_model, | |
LR | |||
) |
Add the momentum-SGD update.
Definition at line 28 of file net_construct_bench.py.