gptmed 0.4.0__py3-none-any.whl → 0.4.1__py3-none-any.whl

gptmed/api.py

@@ -203,6 +203,7 @@ def train_from_config(
         log_dir=args['log_dir'],
         device=actual_device,  # Use actual device from DeviceManager
         seed=args['seed'],
+        resume_from=args.get('resume_from'),  # Pass resume checkpoint path
     )
     
     # Create optimizer

gptmed/configs/train_config.py

@@ -37,6 +37,7 @@ COMMON FAILURE MODES:
 
 from dataclasses import dataclass
 from pathlib import Path
+from typing import Optional
 import json
 
 
@@ -91,6 +92,9 @@ class TrainingConfig:
     # Reproducibility
     seed: int = 42
 
+    # Resume training
+    resume_from: Optional[str] = None  # Path to checkpoint to resume from
+
     def to_dict(self) -> dict:
         """Convert to dictionary."""
         return {
@@ -117,6 +121,7 @@ class TrainingConfig:
             "device": self.device,
             "use_amp": self.use_amp,
             "seed": self.seed,
+            "resume_from": self.resume_from,
         }
 
     def save(self, path: Path):

{gptmed-0.4.0.dist-info → gptmed-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gptmed
-Version: 0.4.0
+Version: 0.4.1
 Summary: A lightweight GPT-based language model framework for training custom question-answering models on any domain
 Author-email: Sanjog Sigdel <sigdelsanjog@gmail.com>
 Maintainer-email: Sanjog Sigdel <sigdelsanjog@gmail.com>

{gptmed-0.4.0.dist-info → gptmed-0.4.1.dist-info}/RECORD

@@ -1,8 +1,8 @@
 gptmed/__init__.py,sha256=lSCUt0jmB81dEG0UroQdrk8TMG9Hv-_a14nAvB6yYiQ,2725
-gptmed/api.py,sha256=k9a_1F2h__xgKnH2l0FaJqAqu-iTYt5tu_VfVO0UhrA,9806
+gptmed/api.py,sha256=GbOaFYvDQgn_qm2FCLiiSXDp-uhvlUNRcY6zTFmsBAs,9882
 gptmed/configs/__init__.py,sha256=yRa-zgPQ-OCzu8fvCrfWMG-CjF3dru3PZzknzm0oUaQ,23
 gptmed/configs/config_loader.py,sha256=3GQ1iCNpdJ5yALWXA3SPPHRkaUO-117vdArEL6u7sK8,6354
-gptmed/configs/train_config.py,sha256=KqfNBh9hdTTd_6gEAlrClU8sVFSlVDmZJOrf3cPwFe8,4657
+gptmed/configs/train_config.py,sha256=cuGE5o4N3TA65Sue8J3XrbmI5QKI7Ww3WeHd2M7yoHQ,4828
 gptmed/configs/training_config.yaml,sha256=EEZZa3kcsZr3g-_fKDPYZt4_NTpmS-3NvJrTYSWNc8g,2874
 gptmed/data/__init__.py,sha256=iAHeakB5pBAd7MkmarPPY0UKS9bTaO_winLZ23Y2O90,54
 gptmed/data/parsers/__init__.py,sha256=BgVzXuZgeE5DUCC4SzN7vflL40wQ4Q4_4DmJ1Y43_nw,211
@@ -40,9 +40,9 @@ gptmed/training/utils.py,sha256=pJxCwneNr2STITIYwIDCxRzIICDFOxOMzK8DT7ck2oQ,5651
 gptmed/utils/__init__.py,sha256=XuMhIqOXF7mjnog_6Iky-hSbwvFb0iK42B4iDUpgi0U,44
 gptmed/utils/checkpoints.py,sha256=jPKJtO0YRZieGmpwqotgDkBzd__s_raDxS1kLpfjBJE,7113
 gptmed/utils/logging.py,sha256=7dJc1tayMxCBjFSDXe4r9ACUTpoPTTGsJ0UZMTqZIDY,5303
-gptmed-0.4.0.dist-info/licenses/LICENSE,sha256=v2spsd7N1pKFFh2G8wGP_45iwe5S0DYiJzG4im8Rupc,1066
-gptmed-0.4.0.dist-info/METADATA,sha256=kVsL6zbBoGw1jrlaDiPkBAr_D7YedPCSwZkjGCFz04c,13832
-gptmed-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gptmed-0.4.0.dist-info/entry_points.txt,sha256=ATqOzTtPVdUiFX5ZSeo3n9JkUCqocUxEXTgy1CfNRZE,110
-gptmed-0.4.0.dist-info/top_level.txt,sha256=mhyEq3rG33t21ziJz5w3TPgx0RjPf4zXMNUx2JTiNmE,7
-gptmed-0.4.0.dist-info/RECORD,,
+gptmed-0.4.1.dist-info/licenses/LICENSE,sha256=v2spsd7N1pKFFh2G8wGP_45iwe5S0DYiJzG4im8Rupc,1066
+gptmed-0.4.1.dist-info/METADATA,sha256=ek6fOyXkyu9k_d66_Vt2Rn8-ZyikKn-raTkdeXFIEGg,13832
+gptmed-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gptmed-0.4.1.dist-info/entry_points.txt,sha256=ATqOzTtPVdUiFX5ZSeo3n9JkUCqocUxEXTgy1CfNRZE,110
+gptmed-0.4.1.dist-info/top_level.txt,sha256=mhyEq3rG33t21ziJz5w3TPgx0RjPf4zXMNUx2JTiNmE,7
+gptmed-0.4.1.dist-info/RECORD,,