OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
a53192e334
pytorch/caffe2/python/modeling
History
Tao Wu b9e64ecef1 allow param_info to set optimizer 
Summary: this diff adds optimizer into param_info, and the associated implementations for modelhelper and brew to set optimizer for each individual parameter.

Reviewed By: kennyhorror

Differential Revision: D5385432

fbshipit-source-id: 5d682f9d1ab077e04a5d76a24d71470f4e64fc92
2017-07-12 08:49:48 -07:00
..
initializers_test.py Skip fp16 initializer test for CPU-only builds 2017-06-19 12:21:25 -07:00
initializers.py Create ParameterSharing abstraction for Caffe2. 2017-06-05 11:49:54 -07:00
parameter_info.py allow param_info to set optimizer 2017-07-12 08:49:48 -07:00
parameter_sharing_test.py Create ParameterSharing abstraction for Caffe2. 2017-06-05 11:49:54 -07:00
parameter_sharing.py Create ParameterSharing abstraction for Caffe2. 2017-06-05 11:49:54 -07:00