Merge pull request #25 from levelfour/fix-map-update

MutableMapping.update may cause problems?
moskomule · Oct 4, 2019 · 59eab55 · 59eab55
2 parents 3f3b880 + c3be224
commit 59eab55
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/homura/trainers.py b/homura/trainers.py
@@ -188,7 +188,8 @@ def _iteration(self,
             results = dict(loss=loss, output=output)
             self._iteration_map.update(**results)
         else:
-            self._iteration_map.update(**results)
+            for k in results:
+                self._iteration_map[k] = results[k]
         self._iteration_map[DATA] = data
         with torch.no_grad():
             self._callbacks.after_iteration(self._iteration_map)