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.
1.8.11