hqde 0.1.3__py3-none-any.whl → 0.1.4__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.

Potentially problematic release.

This version of hqde might be problematic. Click here for more details.

Files changed (6) hide show
  1. hqde/core/hqde_system.py +5 -1
  2. {hqde-0.1.3.dist-info → hqde-0.1.4.dist-info}/METADATA +1 -1
  3. {hqde-0.1.3.dist-info → hqde-0.1.4.dist-info}/RECORD +6 -6
  4. {hqde-0.1.3.dist-info → hqde-0.1.4.dist-info}/WHEEL +0 -0
  5. {hqde-0.1.3.dist-info → hqde-0.1.4.dist-info}/licenses/LICENSE +0 -0
  6. {hqde-0.1.3.dist-info → hqde-0.1.4.dist-info}/top_level.txt +0 -0
hqde/core/hqde_system.py CHANGED
@@ -153,7 +153,11 @@ class DistributedEnsembleManager:
153
153
 
154
154
  def create_ensemble_workers(self, model_class, model_kwargs: Dict[str, Any]):
155
155
  """Create distributed ensemble workers."""
156
- @ray.remote
156
+ # Calculate GPU fraction per worker (divide available GPUs among workers)
157
+ num_gpus = torch.cuda.device_count() if torch.cuda.is_available() else 0
158
+ gpu_per_worker = num_gpus / self.num_workers if num_gpus > 0 else 0
159
+
160
+ @ray.remote(num_gpus=gpu_per_worker)
157
161
  class EnsembleWorker:
158
162
  def __init__(self, model_class, model_kwargs):
159
163
  self.model = model_class(**model_kwargs)
{hqde-0.1.3.dist-info → hqde-0.1.4.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: hqde
3
- Version: 0.1.3
3
+ Version: 0.1.4
4
4
  Summary: Hierarchical Quantum-Distributed Ensemble Learning Framework
5
5
  Author-email: HQDE Team <hqde@example.com>
6
6
  Maintainer-email: HQDE Team <hqde@example.com>
{hqde-0.1.3.dist-info → hqde-0.1.4.dist-info}/RECORD RENAMED
@@ -2,7 +2,7 @@ hqde/__init__.py,sha256=jxetUxE9gTqHOpxYDx2ZwcJKIkHa7eMIprl9dGuqiBI,1353
2
2
  hqde/__main__.py,sha256=6Dozsi53MxYGWL_vFJaH4KuTVJu_RtcD0Tjpn1bGiF0,3054
3
3
  hqde/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
4
  hqde/core/__init__.py,sha256=ZLB6uBaJKyfTaSeHckiyW21HUzKcDGo52hdj0gJzL1U,499
5
- hqde/core/hqde_system.py,sha256=71lRH7z-7_FIvc8_l-OgjwX6ordGDjtPemYg44Wn3cI,19578
5
+ hqde/core/hqde_system.py,sha256=jrT4tlz8eusSf-2EkTCEvloXvZYjOUfMl8NZBmBoGYc,19851
6
6
  hqde/distributed/__init__.py,sha256=qOzxRxTJejXGiNwv2Ibts5m4pSLt8KtzLWu0RgEQnuU,584
7
7
  hqde/distributed/fault_tolerance.py,sha256=TMfLCXL14BO0TYL834r41oKoZ9dxxTp99Ux1d6hBMfw,14801
8
8
  hqde/distributed/hierarchical_aggregator.py,sha256=UbtB2qU1ws70594woK_bJhvbjN6PA9XAWxggT8F00rY,15790
@@ -17,8 +17,8 @@ hqde/utils/config_manager.py,sha256=GY_uFBwj6qJ_ESkopIjR_vQwLIcILNqdNj2o_GFFAdg,
17
17
  hqde/utils/data_utils.py,sha256=2CVHULh45Usf9zcvM7i3qeZkpLNzRSEPDQ4vCjHk14E,264
18
18
  hqde/utils/performance_monitor.py,sha256=J4VntvwnBwMRAArtuVDr13oKcVjr4y5WWowW1dm21rI,16644
19
19
  hqde/utils/visualization.py,sha256=NwiUrgMQFBeqrIblp2qFWl71bFNG58FZKESK2-GB8eM,185
20
- hqde-0.1.3.dist-info/licenses/LICENSE,sha256=ACTIUEzMwldWiL-H94KKJaGyUNxu_L5EQylXnagPamE,1065
21
- hqde-0.1.3.dist-info/METADATA,sha256=m-9HxUIMbxsadqKAbEx9VHVB88xvEOGHM2IiM3woqLM,7887
22
- hqde-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
23
- hqde-0.1.3.dist-info/top_level.txt,sha256=lDNw5jGWRhvYQohaYu7Cm4F7vd3YFPIwoLULxJNopqc,5
24
- hqde-0.1.3.dist-info/RECORD,,
20
+ hqde-0.1.4.dist-info/licenses/LICENSE,sha256=ACTIUEzMwldWiL-H94KKJaGyUNxu_L5EQylXnagPamE,1065
21
+ hqde-0.1.4.dist-info/METADATA,sha256=aQjNKhBr3StdtoB_rr5J4RSIpNtdiOeiOD1GstI6chg,7887
22
+ hqde-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
23
+ hqde-0.1.4.dist-info/top_level.txt,sha256=lDNw5jGWRhvYQohaYu7Cm4F7vd3YFPIwoLULxJNopqc,5
24
+ hqde-0.1.4.dist-info/RECORD,,
{hqde-0.1.3.dist-info → hqde-0.1.4.dist-info}/WHEEL RENAMED
File without changes