d9d 0.2.0__tar.gz → 0.2.1__tar.gz

{d9d-0.2.0 → d9d-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: d9d
-Version: 0.2.0
+Version: 0.2.1
 Summary: d9d - d[istribute]d - distributed training framework based on PyTorch that tries to be efficient yet hackable
 License: Apache-2.0
 License-File: LICENSE

{d9d-0.2.0 → d9d-0.2.1}/d9d/loop/component/gradient_manager.py

@@ -78,8 +78,12 @@ class GradientManager:
         self._grads_to_scale = None
 
     def _scale_grads(self):
+        if self._grads_to_scale is None:
+            raise ValueError("You should bind the manager first.")
+
         scale_factor = 1.0 / self._loss.accumulated_weight
-        torch._foreach_mul_(self._grads_to_scale, scale_factor)
+        if len(self._grads_to_scale) > 0:
+            torch._foreach_mul_(self._grads_to_scale, scale_factor)
 
     @contextmanager
     def install(self):

{d9d-0.2.0 → d9d-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "d9d"
-version = "0.2.0"
+version = "0.2.1"
 description = "d9d - d[istribute]d - distributed training framework based on PyTorch that tries to be efficient yet hackable"
 authors = [
     {name = "Maksim Afanasyev", email = "mr.applexz@gmail.com"}