Merge pull request #24 from moskomule/dev

Minor bugfix
moskomule · Jul 8, 2019 · 3f3b880 · 3f3b880
2 parents be11001 + 58f28b0
commit 3f3b880
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/homura/utils/containers.py b/homura/utils/containers.py
@@ -130,6 +130,6 @@ def load_state_dict(self, state_dicts: dict):
                 self[k].load_state_dict(v)
 
     def zero_grad(self):
-        for v in self.items():
+        for v in self.values():
             if hasattr(v, "zero_grad"):
                 v.zero_grad()
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
     requirements = f.read().split()
 
 setup(name="homura",
-      version="0.6.0",
+      version="0.6.1",
       author="moskomule",
       author_email="hataya@nlab.jp",
       packages=find_packages(exclude=["test", "docs", "examples"]),