Add MER strategy [PEP8 fix]

ContinualAI · Nov 26, 2022 · 9328ad6 · 9328ad6
1 parent 55dc7c7
commit 9328ad6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/avalanche/training/supervised/mer.py b/avalanche/training/supervised/mer.py
@@ -125,17 +125,16 @@ def _inner_updates(self, **kwargs):
             # Within-batch Reptile update
             w_aft_t = self.model.state_dict()
             self.model.load_state_dict(
-                {name: w_bef_t[name] +
-                       ((w_aft_t[name] - w_bef_t[name]) * self.beta)
+                {name: w_bef_t[name] + ((w_aft_t[name] - w_bef_t[name])
+                                        * self.beta)
                  for name in w_bef_t}
             )
 
     def _outer_update(self, **kwargs):
         w_aft = self.model.state_dict()
         self.model.load_state_dict(
-            {name: self.w_bef[name] +
-                   ((w_aft[name] - self.w_bef[name]) * self.gamma)
-             for name in self.w_bef}
+            {name: self.w_bef[name] + ((w_aft[name] - self.w_bef[name])
+                                       * self.gamma) for name in self.w_bef}
         )
         with torch.no_grad():
             pred = self.forward()

diff --git a/avalanche/training/templates/common_templates.py b/avalanche/training/templates/common_templates.py
@@ -450,4 +450,4 @@ def __init__(
         #
         #    This dataset may contain samples from different experiences. If you
         #    want the original data for the current experience
-        #    use :attr:`.BaseTemplate.experience`.
+        #    use :attr:`.BaseTemplate.experience`.