[Fix] ZeroRedundancyOptimizer ambiguous error with param groups when pytorch < 1.12.0 (#818) (4147e976) · Commits · Florian Schiffel / ICV-mmengine_basecode

Unverified Commit 4147e976 authored 2 years ago by Qian Zhao Committed by GitHub 2 years ago

[Fix] ZeroRedundancyOptimizer ambiguous error with param groups when pytorch < 1.12.0 (#818)

* fix zero_optimizer error with param groups when pytorch < 1.12.0

* add docstring

* fix docstring

* add unittest

* change ut to use a valid paramwise_cfg

* modify ut

* fix as comments

parent 89477b5c

No related branches found

No related tags found

Hide whitespace changes

Inline Side-by-side

Showing with 56 additions and 21 deletions

Please register or to comment