From d87192e77fc1031fa7d4565dbf939472048fac48 Mon Sep 17 00:00:00 2001 From: moskomule Date: Mon, 8 Jul 2019 20:45:29 +0900 Subject: [PATCH 1/2] fix StepDict --- homura/utils/containers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/homura/utils/containers.py b/homura/utils/containers.py index 1f8e9cf6..5968ec21 100644 --- 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() From 58f28b0c8aca8134c955b7921d79017bd89d4639 Mon Sep 17 00:00:00 2001 From: moskomule Date: Mon, 8 Jul 2019 22:39:28 +0900 Subject: [PATCH 2/2] update --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 635717a2..3736f2a9 100644 --- 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"]),