d9d 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (5) hide show
  1. d9d/loop/component/gradient_manager.py +5 -1
  2. {d9d-0.2.0.dist-info → d9d-0.2.1.dist-info}/METADATA +1 -1
  3. {d9d-0.2.0.dist-info → d9d-0.2.1.dist-info}/RECORD +5 -5
  4. {d9d-0.2.0.dist-info → d9d-0.2.1.dist-info}/WHEEL +0 -0
  5. {d9d-0.2.0.dist-info → d9d-0.2.1.dist-info}/licenses/LICENSE +0 -0
d9d/loop/component/gradient_manager.py CHANGED
@@ -78,8 +78,12 @@ class GradientManager:
78
78
  self._grads_to_scale = None
79
79
 
80
80
  def _scale_grads(self):
81
+ if self._grads_to_scale is None:
82
+ raise ValueError("You should bind the manager first.")
83
+
81
84
  scale_factor = 1.0 / self._loss.accumulated_weight
82
- torch._foreach_mul_(self._grads_to_scale, scale_factor)
85
+ if len(self._grads_to_scale) > 0:
86
+ torch._foreach_mul_(self._grads_to_scale, scale_factor)
83
87
 
84
88
  @contextmanager
85
89
  def install(self):
{d9d-0.2.0.dist-info → d9d-0.2.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: d9d
3
- Version: 0.2.0
3
+ Version: 0.2.1
4
4
  Summary: d9d - d[istribute]d - distributed training framework based on PyTorch that tries to be efficient yet hackable
5
5
  License: Apache-2.0
6
6
  License-File: LICENSE
{d9d-0.2.0.dist-info → d9d-0.2.1.dist-info}/RECORD RENAMED
@@ -71,7 +71,7 @@ d9d/loop/component/checkpointer.py,sha256=3yhjMutYzeQLxkHYzMe1wpKcNpIeVAv_wtMpQj
71
71
  d9d/loop/component/data_loader_factory.py,sha256=XOI9_XLNmqjMkmQguPM4g_-LbhANNS73hGPE_6fCyOQ,8029
72
72
  d9d/loop/component/garbage_collector.py,sha256=t1wHN4HvaaDfvhnBPfSsq5_7THd_OuYO8tuUA95-UGQ,2867
73
73
  d9d/loop/component/gradient_clipper.py,sha256=0nDGELrwczlqCpHojeC4N3j7dMUhELnod2pWTY1Hv3g,3078
74
- d9d/loop/component/gradient_manager.py,sha256=rtuDooke4HeoBoMXr_YqxSYecZHSzr-pAbXGmSsl10Q,4976
74
+ d9d/loop/component/gradient_manager.py,sha256=AWRxem9TGlmj7ktAKGpPF6YOVbQja839zAfE4blu9Ys,5131
75
75
  d9d/loop/component/job_logger.py,sha256=CZx9EGnILwGQ8mqTaKstjVp_n_5_UO5fQnTEWuJAdsM,5183
76
76
  d9d/loop/component/job_profiler.py,sha256=qTK9LUAbLzuvwfkA_UyOVUMsO5s6cy61x7ZeuWQmiCQ,1835
77
77
  d9d/loop/component/model_stage_exporter.py,sha256=g1f7WDmsJ6MAN_H8ymh4B3MHIURibUhO_3Mvespwfpc,1332
@@ -236,7 +236,7 @@ d9d/tracker/provider/aim/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
236
236
  d9d/tracker/provider/aim/config.py,sha256=7-Kgt2XhdNRhgxLYEWpB27uhgB-RBiPrGvH6S2L3rIM,672
237
237
  d9d/tracker/provider/aim/tracker.py,sha256=g34BdjIYEftEwuK2oTKKzHLN3dpG7-i14zhXIF_onIo,3110
238
238
  d9d/tracker/provider/null.py,sha256=c1nvUaOz8RbRY8XzwSPTi7t0lSsmdlwGAYfYgprwaf8,1440
239
- d9d-0.2.0.dist-info/METADATA,sha256=Uaqgvq05qTQMSGT1QpFzoljNkNYFmFC1S0yfXcl5bM8,6459
240
- d9d-0.2.0.dist-info/WHEEL,sha256=kJCRJT_g0adfAJzTx2GUMmS80rTJIVHRCfG0DQgLq3o,88
241
- d9d-0.2.0.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
242
- d9d-0.2.0.dist-info/RECORD,,
239
+ d9d-0.2.1.dist-info/METADATA,sha256=VP6mlaA7Wg_nhSdDSgZeXfGm7v4uoQiT_Q72ClPIN0M,6459
240
+ d9d-0.2.1.dist-info/WHEEL,sha256=kJCRJT_g0adfAJzTx2GUMmS80rTJIVHRCfG0DQgLq3o,88
241
+ d9d-0.2.1.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
242
+ d9d-0.2.1.dist-info/RECORD,,
{d9d-0.2.0.dist-info → d9d-0.2.1.dist-info}/WHEEL RENAMED
File without changes