Fix optimizer for not supporting all kinds of iterables (#5355)
* added flatten backward
* flatten and softmax backward
* fix bug for not supporting all kinds of iterables in optimizers
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/nn/optimizer/adam.py 2 additions, 2 deletionsoneflow/python/nn/optimizer/adam.py
- oneflow/python/nn/optimizer/adamw.py 2 additions, 2 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 2 additions, 2 deletionsoneflow/python/nn/optimizer/rmsprop.py
- oneflow/python/nn/optimizer/sgd.py 2 additions, 2 deletionsoneflow/python/nn/optimizer/sgd.py
Please register or sign in to comment