Smoothl1loss (#5256)
* implement smoothl1loss * add smoothl1loss test * fix bug * fix bug * remove swp file * fix bug * refine * refine_grpc_cmake_for_conda * refine * refine * refine * rm * refine * refine * refine * refine * also fix opencv * refine * reorder * refine gflags and glog * rm useless * rm useless * rm useless * refine * rm var only available in protobuf * add not-suppported check for reduce and size_average * add test for datatype of int32 * auto format by CI * implement backward and rebuild Factor * rebuild Factor for smooth_l1_loss_grad * auto format by CI * fix bug Co-authored-by:jackalcooper <jackalcooper@gmail.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org>
Showing
- docs/source/experimental.rst 1 addition, 0 deletionsdocs/source/experimental.rst
- oneflow/core/autograd/gradient_funcs/smoothl1_loss.cpp 76 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/smoothl1_loss.cpp
- oneflow/core/functional/functional_api.yaml 9 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/nn_functor.cpp 18 additions, 0 deletionsoneflow/core/functional/impl/nn_functor.cpp
- oneflow/core/functional/impl/nn_grad_functor.cpp 23 additions, 0 deletionsoneflow/core/functional/impl/nn_grad_functor.cpp
- oneflow/python/nn/modules/loss.py 145 additions, 2 deletionsoneflow/python/nn/modules/loss.py
- oneflow/python/test/modules/test_smoothl1loss.py 121 additions, 0 deletionsoneflow/python/test/modules/test_smoothl1loss.py
Please register or sign in to comment