fix: grad clip fn param in AMPOptimizerCallback (#948)

catalyst-team · Sep 29, 2020 · f5aeacc · f5aeacc
1 parent 23307f6
commit f5aeacc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -25,7 +25,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 ### Fixed
 
-- 
+- `AMPOptimizerCallback` - fix grad clip fn support ([#948](https://github.com/catalyst-team/catalyst/pull/948))
+-
 
 
 ## [20.09.1] - 2020-09-25

diff --git a/catalyst/core/callbacks/optimizer.py b/catalyst/core/callbacks/optimizer.py
@@ -307,7 +307,7 @@ def grad_step(
             # Unscales the gradients of
             # optimizer's assigned params in-place
             self.scaler.unscale_(optimizer)
-            for group in zip(optimizer.param_groups):
+            for group in optimizer.param_groups:
                 # Since the gradients of optimizer's
                 # assigned params are unscaled, clips as usual:
                 grad_clip_fn(group["params"])