Support custom parameters for optimizer (#4881)
* feat(Optim): support custom parameters for optimizer
* feat(Adam): adam support custom parameters
* feat(Adamw): adamw support custom parameters
* feat(RMSprop): rmsprop support custom parameters
* style(Optim): refine adam and adamw
Co-authored-by: 
Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
Showing
- oneflow/core/framework/attr_map.cpp 1 addition, 1 deletiononeflow/core/framework/attr_map.cpp
- oneflow/python/nn/optimizer/adam.py 28 additions, 25 deletionsoneflow/python/nn/optimizer/adam.py
- oneflow/python/nn/optimizer/adamw.py 28 additions, 22 deletionsoneflow/python/nn/optimizer/adamw.py
- oneflow/python/nn/optimizer/optimizer.py 2 additions, 2 deletionsoneflow/python/nn/optimizer/optimizer.py
- oneflow/python/nn/optimizer/rmsprop.py 58 additions, 58 deletionsoneflow/python/nn/optimizer/rmsprop.py
- oneflow/python/nn/optimizer/sgd.py 37 additions, 33 deletionsoneflow/python/nn/optimizer/sgd.py
- oneflow/python/test/modules/test_optim_adam.py 10 additions, 9 deletionsoneflow/python/test/modules/test_optim_adam.py
- oneflow/python/test/modules/test_optim_adamw.py 8 additions, 7 deletionsoneflow/python/test/modules/test_optim_adamw.py
- oneflow/python/test/modules/test_optim_rmsprop.py 12 additions, 9 deletionsoneflow/python/test/modules/test_optim_rmsprop.py
- oneflow/python/test/modules/test_optim_sgd.py 1 addition, 4 deletionsoneflow/python/test/modules/test_optim_sgd.py
Please register or sign in to comment