nshtrainer 0.10.17__py3-none-any.whl → 0.10.19__py3-none-any.whl

nshtrainer/_checkpoint/metadata.py

@@ -92,7 +92,7 @@ def _write_checkpoint_metadata(
     except Exception as e:
         log.warning(f"Failed to write metadata to {checkpoint_path}: {e}")
     else:
-        log.info(f"Checkpoint metadata written to {checkpoint_path}")
+        log.debug(f"Checkpoint metadata written to {checkpoint_path}")
 
     # Write the hparams to the checkpoint directory
     try:
@@ -101,7 +101,7 @@ def _write_checkpoint_metadata(
     except Exception as e:
         log.warning(f"Failed to write hparams to {checkpoint_path}: {e}")
     else:
-        log.info(f"Checkpoint metadata written to {checkpoint_path}")
+        log.debug(f"Checkpoint metadata written to {checkpoint_path}")
 
 
 def _remove_checkpoint_metadata(checkpoint_path: Path):
@@ -112,7 +112,7 @@ def _remove_checkpoint_metadata(checkpoint_path: Path):
         except Exception as e:
             log.warning(f"Failed to remove {path}: {e}")
         else:
-            log.info(f"Removed {path}")
+            log.debug(f"Removed {path}")
 
 
 def _link_checkpoint_metadata(checkpoint_path: Path, linked_checkpoint_path: Path):
@@ -133,7 +133,7 @@ def _link_checkpoint_metadata(checkpoint_path: Path, linked_checkpoint_path: Pat
         except Exception as e:
             log.warning(f"Failed to link {path} to {linked_path}: {e}")
         else:
-            log.info(f"Linked {path} to {linked_path}")
+            log.debug(f"Linked {path} to {linked_path}")
 
 
 def _checkpoint_sort_key_fn(key: Callable[[CheckpointMetadata, Path], Any]):

nshtrainer/model/config.py

@@ -315,7 +315,7 @@ class WandbLoggerConfig(CallbackConfigBase, BaseLoggerConfig):
                 if pkg_resources.parse_version(
                     wandb.__version__
                 ) < pkg_resources.parse_version("0.17.5"):
-                    raise ValueError(
+                    log.warning(
                         "The version of WandB installed does not support the `wandb-core` backend "
                         f"(expected version >= 0.17.5, found version {wandb.__version__}). "
                         "Please either upgrade to a newer version of WandB or disable the `use_wandb_core` option."

{nshtrainer-0.10.17.dist-info → nshtrainer-0.10.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 0.10.17
+Version: 0.10.19
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-0.10.17.dist-info → nshtrainer-0.10.19.dist-info}/RECORD

@@ -1,6 +1,6 @@
 nshtrainer/__init__.py,sha256=39loiLLXbaGiozEsAn8mPHopxaPsek8JsgR9DD2gxtY,583
 nshtrainer/_checkpoint/loader.py,sha256=48flPr1XgQHOgIPaCrRqOEvRuG0SZuV3cQ1vgHLqFqI,11025
-nshtrainer/_checkpoint/metadata.py,sha256=GlhlAyJh5gcp3R8l2Y3eAUQtQzBnitFlB0xdx-khEUQ,5579
+nshtrainer/_checkpoint/metadata.py,sha256=3yxGxHLIVwKh5K4L8LYOEK3GQ6HQXy89CGcy9zarApo,5583
 nshtrainer/_checkpoint/saver.py,sha256=z_c7a91O4Bh4lZZjqJgxT3w25qFlJsOopV3cpJtkHk8,1655
 nshtrainer/_experimental/__init__.py,sha256=2tQIcrWT8U8no_AeBTYnozaTmxN40kuAJdGQ4b-PoWM,120
 nshtrainer/_experimental/flops/__init__.py,sha256=edo9Ez3LlrnxkNRX9W6YBhPkRPKYGLpkpnl5gx7sEX8,1550
@@ -52,7 +52,7 @@ nshtrainer/metrics/__init__.py,sha256=ObLIELGguIEcUpRsUkqh1ltrvZii6vglTpJGrPvoy0
 nshtrainer/metrics/_config.py,sha256=hWWS4IXENRyH3RmJ7z1Wx1n3Lt1sNMlGOrcU6PW15o0,1104
 nshtrainer/model/__init__.py,sha256=NpvyQHmGaHB8xdraHmm8l7kDHLmvJSgBNQKkfYqtgyI,1454
 nshtrainer/model/base.py,sha256=AXRfEsFAT0Ln7zjYVPU5NgtHS_c8FZM-M4pyLamO7OA,17516
-nshtrainer/model/config.py,sha256=fcRhk5LYNvD0kptxT_qSA6OO1WRF5ko4NVTeaZvTuKk,54779
+nshtrainer/model/config.py,sha256=4vze6tpMFAgpk532T33jmSH1lHfolHK1vAJEaa2-Vxs,54774
 nshtrainer/model/modules/callback.py,sha256=K0-cyEtBcQhI7Q2e-AGTE8T-GghUPY9DYmneU6ULV6g,6401
 nshtrainer/model/modules/debug.py,sha256=Yy7XEdPou9BkCsD5hJchwJGmCVGrfUru5g9VjPM4uAw,1120
 nshtrainer/model/modules/distributed.py,sha256=ABpR9d-3uBS_fivfy_WYW-dExW6vp5BPaoPQnOudHng,1725
@@ -79,6 +79,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
 nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
 nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
 nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
-nshtrainer-0.10.17.dist-info/METADATA,sha256=yIpZNES4PCMfXUmDIpfa_1xOa3J3PNgdMzu1_OQXq94,696
-nshtrainer-0.10.17.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-0.10.17.dist-info/RECORD,,
+nshtrainer-0.10.19.dist-info/METADATA,sha256=Kxp5ov0XQDog0waNw4vCWIaEP9EMSSMvs7YAPj3DDDU,696
+nshtrainer-0.10.19.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-0.10.19.dist-info/RECORD,,