nshtrainer 0.10.18__py3-none-any.whl → 0.10.19__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.
|
@@ -92,7 +92,7 @@ def _write_checkpoint_metadata(
|
|
|
92
92
|
except Exception as e:
|
|
93
93
|
log.warning(f"Failed to write metadata to {checkpoint_path}: {e}")
|
|
94
94
|
else:
|
|
95
|
-
log.
|
|
95
|
+
log.debug(f"Checkpoint metadata written to {checkpoint_path}")
|
|
96
96
|
|
|
97
97
|
# Write the hparams to the checkpoint directory
|
|
98
98
|
try:
|
|
@@ -101,7 +101,7 @@ def _write_checkpoint_metadata(
|
|
|
101
101
|
except Exception as e:
|
|
102
102
|
log.warning(f"Failed to write hparams to {checkpoint_path}: {e}")
|
|
103
103
|
else:
|
|
104
|
-
log.
|
|
104
|
+
log.debug(f"Checkpoint metadata written to {checkpoint_path}")
|
|
105
105
|
|
|
106
106
|
|
|
107
107
|
def _remove_checkpoint_metadata(checkpoint_path: Path):
|
|
@@ -112,7 +112,7 @@ def _remove_checkpoint_metadata(checkpoint_path: Path):
|
|
|
112
112
|
except Exception as e:
|
|
113
113
|
log.warning(f"Failed to remove {path}: {e}")
|
|
114
114
|
else:
|
|
115
|
-
log.
|
|
115
|
+
log.debug(f"Removed {path}")
|
|
116
116
|
|
|
117
117
|
|
|
118
118
|
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
|
|
|
133
133
|
except Exception as e:
|
|
134
134
|
log.warning(f"Failed to link {path} to {linked_path}: {e}")
|
|
135
135
|
else:
|
|
136
|
-
log.
|
|
136
|
+
log.debug(f"Linked {path} to {linked_path}")
|
|
137
137
|
|
|
138
138
|
|
|
139
139
|
def _checkpoint_sort_key_fn(key: Callable[[CheckpointMetadata, Path], Any]):
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
nshtrainer/__init__.py,sha256=39loiLLXbaGiozEsAn8mPHopxaPsek8JsgR9DD2gxtY,583
|
|
2
2
|
nshtrainer/_checkpoint/loader.py,sha256=48flPr1XgQHOgIPaCrRqOEvRuG0SZuV3cQ1vgHLqFqI,11025
|
|
3
|
-
nshtrainer/_checkpoint/metadata.py,sha256=
|
|
3
|
+
nshtrainer/_checkpoint/metadata.py,sha256=3yxGxHLIVwKh5K4L8LYOEK3GQ6HQXy89CGcy9zarApo,5583
|
|
4
4
|
nshtrainer/_checkpoint/saver.py,sha256=z_c7a91O4Bh4lZZjqJgxT3w25qFlJsOopV3cpJtkHk8,1655
|
|
5
5
|
nshtrainer/_experimental/__init__.py,sha256=2tQIcrWT8U8no_AeBTYnozaTmxN40kuAJdGQ4b-PoWM,120
|
|
6
6
|
nshtrainer/_experimental/flops/__init__.py,sha256=edo9Ez3LlrnxkNRX9W6YBhPkRPKYGLpkpnl5gx7sEX8,1550
|
|
@@ -79,6 +79,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
|
|
|
79
79
|
nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
|
|
80
80
|
nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
|
|
81
81
|
nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
|
|
82
|
-
nshtrainer-0.10.
|
|
83
|
-
nshtrainer-0.10.
|
|
84
|
-
nshtrainer-0.10.
|
|
82
|
+
nshtrainer-0.10.19.dist-info/METADATA,sha256=Kxp5ov0XQDog0waNw4vCWIaEP9EMSSMvs7YAPj3DDDU,696
|
|
83
|
+
nshtrainer-0.10.19.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
84
|
+
nshtrainer-0.10.19.dist-info/RECORD,,
|
|
File without changes
|