Fix typo in clone_module function (#330)

learnables · Apr 21, 2022 · f099ddc · f099ddc
1 parent 36ac4fb
commit f099ddc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/learn2learn/utils/__init__.py b/learn2learn/utils/__init__.py
@@ -136,7 +136,7 @@ def clone_module(module, memo=None):
                 else:
                     cloned = buff.clone()
                     clone._buffers[buffer_key] = cloned
-                    memo[param_ptr] = cloned
+                    memo[buff_ptr] = cloned
 
     # Then, recurse for each submodule
     if hasattr(clone, '_modules'):