Feat module repr (#5486)
* feat(Module): support __repr__ function
* feat(Modules): add extra_repr for some modules
* fix(ConvModule): fix conv module extra_repr
* fix(ActivityModule): raise inplace error to warning
* fix(ModuleRepr): fix module print doctest bug
Co-authored-by: 
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/nn/module.py 49 additions, 0 deletionsoneflow/python/nn/module.py
- oneflow/python/nn/modules/activation.py 60 additions, 2 deletionsoneflow/python/nn/modules/activation.py
- oneflow/python/nn/modules/batchnorm.py 6 additions, 0 deletionsoneflow/python/nn/modules/batchnorm.py
- oneflow/python/nn/modules/container.py 12 additions, 2 deletionsoneflow/python/nn/modules/container.py
- oneflow/python/nn/modules/conv.py 36 additions, 0 deletionsoneflow/python/nn/modules/conv.py
- oneflow/python/nn/modules/flatten.py 3 additions, 0 deletionsoneflow/python/nn/modules/flatten.py
- oneflow/python/nn/modules/linear.py 7 additions, 1 deletiononeflow/python/nn/modules/linear.py
- oneflow/python/nn/modules/normalization.py 5 additions, 0 deletionsoneflow/python/nn/modules/normalization.py
- oneflow/python/nn/modules/padding.py 6 additions, 0 deletionsoneflow/python/nn/modules/padding.py
- oneflow/python/nn/modules/pixelshuffle.py 3 additions, 0 deletionsoneflow/python/nn/modules/pixelshuffle.py
- oneflow/python/nn/modules/pooling.py 24 additions, 0 deletionsoneflow/python/nn/modules/pooling.py
- oneflow/python/nn/modules/upsampling.py 8 additions, 0 deletionsoneflow/python/nn/modules/upsampling.py
Please register or sign in to comment