Fix ones zeros like (#4907)
* feat(xxxLikeOp): ones_like and zeros_like use user op
* fix(Optim): fix learning rate device error bug
* style(*): format codes
* style(*): use int instead of np.int
* test(Optim): add optimizer gpu test (#4908)
Co-authored-by: 
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/framework/check_point_v2.py 1 addition, 1 deletiononeflow/python/framework/check_point_v2.py
- oneflow/python/nn/modules/constant.py 20 additions, 2 deletionsoneflow/python/nn/modules/constant.py
- oneflow/python/nn/optimizer/adam.py 5 additions, 3 deletionsoneflow/python/nn/optimizer/adam.py
- oneflow/python/nn/optimizer/adamw.py 3 additions, 1 deletiononeflow/python/nn/optimizer/adamw.py
- oneflow/python/nn/optimizer/rmsprop.py 3 additions, 1 deletiononeflow/python/nn/optimizer/rmsprop.py
- oneflow/python/nn/optimizer/sgd.py 3 additions, 1 deletiononeflow/python/nn/optimizer/sgd.py
- oneflow/python/test/modules/test_optim_adam.py 14 additions, 3 deletionsoneflow/python/test/modules/test_optim_adam.py
- oneflow/python/test/modules/test_optim_adamw.py 6 additions, 3 deletionsoneflow/python/test/modules/test_optim_adamw.py
- oneflow/python/test/modules/test_optim_rmsprop.py 6 additions, 2 deletionsoneflow/python/test/modules/test_optim_rmsprop.py
- oneflow/python/test/modules/test_optim_sgd.py 6 additions, 3 deletionsoneflow/python/test/modules/test_optim_sgd.py
Please register or sign in to comment