Aapo Kyrola 68cfc52452 MomemtumSGDUpdate -- version of MomentumSGD with update. ...

Summary:
It gives a significant perf boost to do the parameter update inside MomentumSGD, instead of with a separate WeightedSum op.
To ensure backwards compatibility, I made it a separate op.

Also added an unit test.

Reviewed By: prigoyal

Differential Revision: D4262446

fbshipit-source-id: 38e7ee6d7677b398658ac7fe9b7a59b569e033f4

2016-12-15 12:01:29 -08:00

..

lmdb_create_example.py

LMDB example

2016-12-05 11:53:26 -08:00

resnet50_trainer.py

MomemtumSGDUpdate -- version of MomentumSGD with update.

2016-12-15 12:01:29 -08:00