nshtrainer 0.42.0__py3-none-any.whl → 0.43.0__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 (162) hide show
  1. nshtrainer/__init__.py +2 -0
  2. nshtrainer/_callback.py +2 -0
  3. nshtrainer/_checkpoint/loader.py +2 -0
  4. nshtrainer/_checkpoint/metadata.py +2 -0
  5. nshtrainer/_checkpoint/saver.py +2 -0
  6. nshtrainer/_directory.py +4 -2
  7. nshtrainer/_experimental/__init__.py +2 -0
  8. nshtrainer/_hf_hub.py +2 -0
  9. nshtrainer/callbacks/__init__.py +45 -29
  10. nshtrainer/callbacks/_throughput_monitor_callback.py +2 -0
  11. nshtrainer/callbacks/actsave.py +2 -0
  12. nshtrainer/callbacks/base.py +2 -0
  13. nshtrainer/callbacks/checkpoint/__init__.py +6 -2
  14. nshtrainer/callbacks/checkpoint/_base.py +2 -0
  15. nshtrainer/callbacks/checkpoint/best_checkpoint.py +2 -0
  16. nshtrainer/callbacks/checkpoint/last_checkpoint.py +4 -2
  17. nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py +6 -2
  18. nshtrainer/callbacks/debug_flag.py +2 -0
  19. nshtrainer/callbacks/directory_setup.py +4 -2
  20. nshtrainer/callbacks/early_stopping.py +6 -4
  21. nshtrainer/callbacks/ema.py +5 -3
  22. nshtrainer/callbacks/finite_checks.py +3 -1
  23. nshtrainer/callbacks/gradient_skipping.py +6 -4
  24. nshtrainer/callbacks/interval.py +2 -0
  25. nshtrainer/callbacks/log_epoch.py +13 -1
  26. nshtrainer/callbacks/norm_logging.py +4 -2
  27. nshtrainer/callbacks/print_table.py +3 -1
  28. nshtrainer/callbacks/rlp_sanity_checks.py +4 -2
  29. nshtrainer/callbacks/shared_parameters.py +4 -2
  30. nshtrainer/callbacks/throughput_monitor.py +2 -0
  31. nshtrainer/callbacks/timer.py +5 -3
  32. nshtrainer/callbacks/wandb_upload_code.py +4 -2
  33. nshtrainer/callbacks/wandb_watch.py +4 -2
  34. nshtrainer/config/__init__.py +130 -90
  35. nshtrainer/config/_checkpoint/loader/__init__.py +10 -8
  36. nshtrainer/config/_checkpoint/metadata/__init__.py +6 -4
  37. nshtrainer/config/_directory/__init__.py +9 -3
  38. nshtrainer/config/_hf_hub/__init__.py +6 -4
  39. nshtrainer/config/callbacks/__init__.py +82 -42
  40. nshtrainer/config/callbacks/actsave/__init__.py +4 -2
  41. nshtrainer/config/callbacks/base/__init__.py +2 -0
  42. nshtrainer/config/callbacks/checkpoint/__init__.py +6 -4
  43. nshtrainer/config/callbacks/checkpoint/_base/__init__.py +6 -4
  44. nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py +2 -0
  45. nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py +6 -4
  46. nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py +6 -4
  47. nshtrainer/config/callbacks/debug_flag/__init__.py +6 -4
  48. nshtrainer/config/callbacks/directory_setup/__init__.py +7 -5
  49. nshtrainer/config/callbacks/early_stopping/__init__.py +9 -7
  50. nshtrainer/config/callbacks/ema/__init__.py +5 -3
  51. nshtrainer/config/callbacks/finite_checks/__init__.py +7 -5
  52. nshtrainer/config/callbacks/gradient_skipping/__init__.py +7 -5
  53. nshtrainer/config/callbacks/norm_logging/__init__.py +9 -5
  54. nshtrainer/config/callbacks/print_table/__init__.py +7 -5
  55. nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py +7 -5
  56. nshtrainer/config/callbacks/shared_parameters/__init__.py +7 -5
  57. nshtrainer/config/callbacks/throughput_monitor/__init__.py +6 -4
  58. nshtrainer/config/callbacks/timer/__init__.py +9 -5
  59. nshtrainer/config/callbacks/wandb_upload_code/__init__.py +7 -5
  60. nshtrainer/config/callbacks/wandb_watch/__init__.py +9 -5
  61. nshtrainer/config/loggers/__init__.py +18 -10
  62. nshtrainer/config/loggers/_base/__init__.py +2 -0
  63. nshtrainer/config/loggers/csv/__init__.py +2 -0
  64. nshtrainer/config/loggers/tensorboard/__init__.py +2 -0
  65. nshtrainer/config/loggers/wandb/__init__.py +18 -10
  66. nshtrainer/config/lr_scheduler/__init__.py +2 -0
  67. nshtrainer/config/lr_scheduler/_base/__init__.py +2 -0
  68. nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py +2 -0
  69. nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py +6 -4
  70. nshtrainer/config/metrics/__init__.py +2 -0
  71. nshtrainer/config/metrics/_config/__init__.py +2 -0
  72. nshtrainer/config/model/__init__.py +8 -6
  73. nshtrainer/config/model/base/__init__.py +4 -2
  74. nshtrainer/config/model/config/__init__.py +8 -6
  75. nshtrainer/config/model/mixins/logger/__init__.py +2 -0
  76. nshtrainer/config/nn/__init__.py +16 -14
  77. nshtrainer/config/nn/mlp/__init__.py +2 -0
  78. nshtrainer/config/nn/nonlinearity/__init__.py +26 -24
  79. nshtrainer/config/optimizer/__init__.py +2 -0
  80. nshtrainer/config/profiler/__init__.py +2 -0
  81. nshtrainer/config/profiler/_base/__init__.py +2 -0
  82. nshtrainer/config/profiler/advanced/__init__.py +6 -4
  83. nshtrainer/config/profiler/pytorch/__init__.py +6 -4
  84. nshtrainer/config/profiler/simple/__init__.py +6 -4
  85. nshtrainer/config/runner/__init__.py +2 -0
  86. nshtrainer/config/trainer/_config/__init__.py +43 -39
  87. nshtrainer/config/trainer/checkpoint_connector/__init__.py +2 -0
  88. nshtrainer/config/util/_environment_info/__init__.py +20 -18
  89. nshtrainer/config/util/config/__init__.py +2 -0
  90. nshtrainer/config/util/config/dtype/__init__.py +2 -0
  91. nshtrainer/config/util/config/duration/__init__.py +2 -0
  92. nshtrainer/data/__init__.py +2 -0
  93. nshtrainer/data/balanced_batch_sampler.py +2 -0
  94. nshtrainer/data/datamodule.py +2 -0
  95. nshtrainer/data/transform.py +2 -0
  96. nshtrainer/ll/__init__.py +2 -0
  97. nshtrainer/ll/_experimental.py +2 -0
  98. nshtrainer/ll/actsave.py +2 -0
  99. nshtrainer/ll/callbacks.py +2 -0
  100. nshtrainer/ll/config.py +2 -0
  101. nshtrainer/ll/data.py +2 -0
  102. nshtrainer/ll/log.py +2 -0
  103. nshtrainer/ll/lr_scheduler.py +2 -0
  104. nshtrainer/ll/model.py +2 -0
  105. nshtrainer/ll/nn.py +2 -0
  106. nshtrainer/ll/optimizer.py +2 -0
  107. nshtrainer/ll/runner.py +2 -0
  108. nshtrainer/ll/snapshot.py +2 -0
  109. nshtrainer/ll/snoop.py +2 -0
  110. nshtrainer/ll/trainer.py +2 -0
  111. nshtrainer/ll/typecheck.py +2 -0
  112. nshtrainer/ll/util.py +2 -0
  113. nshtrainer/loggers/__init__.py +2 -0
  114. nshtrainer/loggers/_base.py +2 -0
  115. nshtrainer/loggers/csv.py +2 -0
  116. nshtrainer/loggers/tensorboard.py +2 -0
  117. nshtrainer/loggers/wandb.py +6 -4
  118. nshtrainer/lr_scheduler/__init__.py +2 -0
  119. nshtrainer/lr_scheduler/_base.py +2 -0
  120. nshtrainer/lr_scheduler/linear_warmup_cosine.py +2 -0
  121. nshtrainer/lr_scheduler/reduce_lr_on_plateau.py +2 -0
  122. nshtrainer/metrics/__init__.py +2 -0
  123. nshtrainer/metrics/_config.py +2 -0
  124. nshtrainer/model/__init__.py +2 -0
  125. nshtrainer/model/base.py +2 -0
  126. nshtrainer/model/config.py +2 -0
  127. nshtrainer/model/mixins/callback.py +2 -0
  128. nshtrainer/model/mixins/logger.py +2 -0
  129. nshtrainer/nn/__init__.py +2 -0
  130. nshtrainer/nn/mlp.py +2 -0
  131. nshtrainer/nn/module_dict.py +2 -0
  132. nshtrainer/nn/module_list.py +2 -0
  133. nshtrainer/nn/nonlinearity.py +2 -0
  134. nshtrainer/optimizer.py +2 -0
  135. nshtrainer/profiler/__init__.py +2 -0
  136. nshtrainer/profiler/_base.py +2 -0
  137. nshtrainer/profiler/advanced.py +2 -0
  138. nshtrainer/profiler/pytorch.py +2 -0
  139. nshtrainer/profiler/simple.py +2 -0
  140. nshtrainer/runner.py +2 -0
  141. nshtrainer/scripts/find_packages.py +2 -0
  142. nshtrainer/trainer/__init__.py +2 -0
  143. nshtrainer/trainer/_config.py +16 -13
  144. nshtrainer/trainer/_runtime_callback.py +2 -0
  145. nshtrainer/trainer/checkpoint_connector.py +2 -0
  146. nshtrainer/trainer/signal_connector.py +2 -0
  147. nshtrainer/trainer/trainer.py +2 -0
  148. nshtrainer/util/_environment_info.py +2 -0
  149. nshtrainer/util/bf16.py +2 -0
  150. nshtrainer/util/config/__init__.py +2 -0
  151. nshtrainer/util/config/dtype.py +2 -0
  152. nshtrainer/util/config/duration.py +2 -0
  153. nshtrainer/util/environment.py +2 -0
  154. nshtrainer/util/path.py +2 -0
  155. nshtrainer/util/seed.py +2 -0
  156. nshtrainer/util/slurm.py +3 -0
  157. nshtrainer/util/typed.py +2 -0
  158. nshtrainer/util/typing_utils.py +2 -0
  159. {nshtrainer-0.42.0.dist-info → nshtrainer-0.43.0.dist-info}/METADATA +1 -1
  160. nshtrainer-0.43.0.dist-info/RECORD +162 -0
  161. nshtrainer-0.42.0.dist-info/RECORD +0 -162
  162. {nshtrainer-0.42.0.dist-info → nshtrainer-0.43.0.dist-info}/WHEEL +0 -0
@@ -1,3 +1,5 @@
1
+ from __future__ import annotations
2
+
1
3
  from typing import TYPE_CHECKING
2
4
 
3
5
  from typing_extensions import TypeVar
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.42.0
3
+ Version: 0.43.0
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
@@ -0,0 +1,162 @@
1
+ nshtrainer/__init__.py,sha256=KgAwRuVmetgBOeEfG_09sJ6JO-F3LH5QsBRDiVdNm6o,678
2
+ nshtrainer/_callback.py,sha256=qPvUK3cZZyin9WNWXdKLd4RxqMbaCzyrqnhXdQMIRYg,1049
3
+ nshtrainer/_checkpoint/loader.py,sha256=ViXEr7OPyekNE4FgbBpMkS6ilY7QGLWKtiyTZaqQunc,14194
4
+ nshtrainer/_checkpoint/metadata.py,sha256=4WCXT4nosS4_P1xurZ4dhadSP7iVLB8qyMvCq8rsu8A,4773
5
+ nshtrainer/_checkpoint/saver.py,sha256=LOP8jjKF0Dw9x9H-BKrLMWlEp1XTan2DUK0zQUCWw5U,1360
6
+ nshtrainer/_directory.py,sha256=p2uk1FnISFEpMqlDevKhoWhQsCEtvHUPg459K-86QA8,3053
7
+ nshtrainer/_experimental/__init__.py,sha256=1eAD28upA3EEHsxKKkNXMdl78IML55CsMtiDYF0tsJc,117
8
+ nshtrainer/_hf_hub.py,sha256=JO6Hup2BM0GV6s65LvzAOFtT1u7zmO-mFcAyKaOHYnk,14141
9
+ nshtrainer/callbacks/__init__.py,sha256=KCdmHAXyqiWqdjbGuHnuiqSTZFQgMtZcghRNCV9uuAY,3815
10
+ nshtrainer/callbacks/_throughput_monitor_callback.py,sha256=5-3pk1S2NzJBfdAhlxVLVsoJsewXP0W47udmRz34r_0,23220
11
+ nshtrainer/callbacks/actsave.py,sha256=VY85oImVcwIasvBmF5qX8l_K1lZ88yJTW3zJYqK_0zw,4077
12
+ nshtrainer/callbacks/base.py,sha256=AWuwxpqg-l2PNQjGbxlOwRxRPItLp84t6n-2a6eoIZk,3544
13
+ nshtrainer/callbacks/checkpoint/__init__.py,sha256=SlXrYBF6Sb1ToH5O6P7HcbJBedqJGsEdISlouWVRD3Y,604
14
+ nshtrainer/callbacks/checkpoint/_base.py,sha256=wcrRcFIc59OWBLSGYrKNc6XjkVwAs0ono4P2zFg8keQ,6251
15
+ nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=SHsZDRGOzrEsAWLWlbYB2-bLB-E6vPbDXwNSOOacs1E,2589
16
+ nshtrainer/callbacks/checkpoint/last_checkpoint.py,sha256=79Eh1JX7XePp-Jix-ERe-4-tYJE55P-crF41GhWz5pU,1154
17
+ nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=1Y_76X2SpqBDbILfJKW7pZK0OWLyUhQ1y23qEOUwEE8,3446
18
+ nshtrainer/callbacks/debug_flag.py,sha256=rUQZUfUURHS7Nti9STyzM5ad9kRv_nZjISiR4UU38hU,2047
19
+ nshtrainer/callbacks/directory_setup.py,sha256=k8vpaR_T2TRgZLlB6GAt68LYIzzp-mBAwy4YgVw6ATM,2681
20
+ nshtrainer/callbacks/early_stopping.py,sha256=xM2V7-1JXb7aULQx5EZOOR7gJGlFBdu38_FXSDeADIc,4706
21
+ nshtrainer/callbacks/ema.py,sha256=A64Ah9OUaEi4pj1u6af4Ds5-X0MuW1wu9oVBUdN7qzQ,12191
22
+ nshtrainer/callbacks/finite_checks.py,sha256=EImZPPdS_0VcbVb_REXSYqYl1twPRygreDVeBt_uPAY,2121
23
+ nshtrainer/callbacks/gradient_skipping.py,sha256=lNLhYN4MTZNhI5DER0SLk8NLXrAZw8uPlYAtYVKW5K0,3461
24
+ nshtrainer/callbacks/interval.py,sha256=UCzUzt3XCFVyQyCWL9lOrStkkxesvduNOYk8yMrGTTk,8116
25
+ nshtrainer/callbacks/log_epoch.py,sha256=3bBA0wJMd2sDe_0nYDznOYhcBqdaTz6ua7HlY0AVN6Y,1393
26
+ nshtrainer/callbacks/norm_logging.py,sha256=M81V_DVrGDCcep1UN0fqn1ZC5ZH7biSXjiSdNjZUvf4,6326
27
+ nshtrainer/callbacks/print_table.py,sha256=1o0dMG9CwpBdEGQeG61uwc9QBA692Fy_evwwQJoFq9g,2881
28
+ nshtrainer/callbacks/rlp_sanity_checks.py,sha256=4UnHffbkgT1bCHRP9-KVdbQ2CGGbGvg5zEcUhASW0FU,9987
29
+ nshtrainer/callbacks/shared_parameters.py,sha256=mxa0BseYBj_R-sFZpe-omrIgRlwcQl0766A8v61gD9M,2974
30
+ nshtrainer/callbacks/throughput_monitor.py,sha256=kfM--rVUAm1x9InNMuRcm0jkGN6ecRxSG2w7-NDh3XA,1874
31
+ nshtrainer/callbacks/timer.py,sha256=cf4iHdL-W7w6AddJ2L7VmlOqhCULRaYQY-zuKn6hruQ,4667
32
+ nshtrainer/callbacks/wandb_upload_code.py,sha256=ZQgcm10QuP6wWPIJcLOraoZOPZkl3jfnmEiJsFQ24XA,2350
33
+ nshtrainer/callbacks/wandb_watch.py,sha256=x-DGaDcXpBBuN09QumeHmvGCbTm_IBEesn9yPSyhmTo,2973
34
+ nshtrainer/config/__init__.py,sha256=3fzl30I9Dj9ycFwhGhnx4oFyDiGMUFBhJKWKCVpDdb0,22695
35
+ nshtrainer/config/_checkpoint/loader/__init__.py,sha256=LscuP8J9zSUfjWxKl-uryJzIjQO4_kRkOcFTp868KIk,2399
36
+ nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=X9KxpcoHQbJp6-MTGvp4pct-MYHaHcl82s9yqZ5KiSk,867
37
+ nshtrainer/config/_directory/__init__.py,sha256=w9xYSZRgWq4lp5iBYa3tX5MWtQe5tpYdNSVxv9-nM9k,1029
38
+ nshtrainer/config/_hf_hub/__init__.py,sha256=fXSaOru7AIgIHXKLA2jnPiXyivUdECu7Y5jN5qmTrqk,1063
39
+ nshtrainer/config/callbacks/__init__.py,sha256=K-8apQVhVGh-07wNIkLRqmpWunXvgatf72qnWcnx3WI,7338
40
+ nshtrainer/config/callbacks/actsave/__init__.py,sha256=AkVWS9vCcDJFpPUpyc7i9cjaFZU2kKxDyFDqakMZA-E,809
41
+ nshtrainer/config/callbacks/base/__init__.py,sha256=OdtHDMkYC_ioCEAkg7bSQi3o7e2t5WHPcFjavXdfdTA,602
42
+ nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=TB7AE7X24S3g0xiO4aD0Uzg-pDLlTI7damfR2o83o9k,2700
43
+ nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=rYQ7hPaqv606sY42qjF-9mwBZZdH0T-rorpFRz2IWbA,1255
44
+ nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=dkGFUTuDSppQ__xYggqBPw4a08p0DaiSX0y1dvncHHY,1637
45
+ nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=jrX5HHyoxGPnzlsSq_a2OV6ENXMSRc7aK-LU31818k4,1355
46
+ nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=unh9ao4mJ64aY0k0JXCdVBYyk1_cU5lYqlKOH0nQbO0,1069
47
+ nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=RQiUwWeaR8pcXKGYP5oZnnYALLtdRallgLI6VYB96Ao,908
48
+ nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=dixNsxyeVulEcMqBZKdolZDcl3R8jXhubMQ_Y8PhJX8,965
49
+ nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=NYs90lfyKD_amIM_-u2eJ8CH6UYKLfR-4udmJu18huA,1198
50
+ nshtrainer/config/callbacks/ema/__init__.py,sha256=WpSohYCGplaRYDeXPPoAn-CnafGYw3mRo89gZQp4FnA,809
51
+ nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=VuLUoTMLgQPa3R9mxkkZRIyAeXQyVdC9eNVUhMzmsek,949
52
+ nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=bphb_C91L6RPhUWcVIX_IxBxgFvGkuZD8TVDsPqFywU,981
53
+ nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=x654QELJyLdxlDyt7YrSzVn1-hFNbphkiK4kn2kFbr4,941
54
+ nshtrainer/config/callbacks/print_table/__init__.py,sha256=Z_7QFur2da8V14v5w0YpJldXj-P-ypp59Uta-MMXS5A,961
55
+ nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256=RiTOGnct1THb8jI_reSL12HELMQZPanveJUsLSWkuTc,977
56
+ nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=xNlohJ4UFV-Ojwbh3p0G8o0onbZeTEjpOe_boXUnQmw,981
57
+ nshtrainer/config/callbacks/throughput_monitor/__init__.py,sha256=9pxqThqXOi_RZiDWaHGVN4cLbyxdAZG0toEdKdJVr2M,957
58
+ nshtrainer/config/callbacks/timer/__init__.py,sha256=j1_vVpn7U9A8t6EqkV0_1OFAJZLxeV2rZBOZB-bX4tE,892
59
+ nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=0QEHuBYLjyEvCOzTg6tDAKyHnYyzc4k94PtmOBUsisM,977
60
+ nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=LE7PAsy5r1PttMDsKFYkmGNxgDrKV7ThhcQj5zqrW3k,933
61
+ nshtrainer/config/loggers/__init__.py,sha256=ZLq0EzVXymNryGr_yLrvrzTZF07cRVkRpGD6WxGeIu4,2290
62
+ nshtrainer/config/loggers/_base/__init__.py,sha256=CFo8APn-VRiUI19lcSV97icCJrKefQ1wAFuZTWFn_fQ,562
63
+ nshtrainer/config/loggers/csv/__init__.py,sha256=AgKd0slcqBzfvTX-OkpF_y-EK8zZdR4XX_Ye01DPvT4,755
64
+ nshtrainer/config/loggers/tensorboard/__init__.py,sha256=sw2lvaAJLxFQze5fxY-WJfJHHGZ2RX3VpZ-oqW9a_Ck,896
65
+ nshtrainer/config/loggers/wandb/__init__.py,sha256=xjoA8voAjFRqraj0O6Aktx8mazh5iHAHqJXYsc5SZKo,1602
66
+ nshtrainer/config/lr_scheduler/__init__.py,sha256=f2Rkja13ciy7T2DWitjoVr-J9AuZ1oB6oohBUZ8Y25s,2149
67
+ nshtrainer/config/lr_scheduler/_base/__init__.py,sha256=u9bOJOs_BrZLTg-aUZpfsNDvtE6957KsCgfoV2IuvjA,639
68
+ nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py,sha256=7l9zhLnkX-QlfMdmv0x0_HPxo7SplH6NkeQNRThEphM,1341
69
+ nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py,sha256=CN_4VtFGk7QzhOpRAo-sQ8QqGLg_n4cPxkH6gE6dzDQ,1265
70
+ nshtrainer/config/metrics/__init__.py,sha256=W_nAvsNNCdwGKbYMer93GO2fei9LJacmlRiRkKbLjQI,568
71
+ nshtrainer/config/metrics/_config/__init__.py,sha256=SX9sHbKqloCw5jGv1zS-JmF3M5AaXpdpxJZNE0SPTes,550
72
+ nshtrainer/config/model/__init__.py,sha256=0Hk6gJbeNqiQpk3A729724fBFKZ_xYkDWadlEwg-K4Y,1561
73
+ nshtrainer/config/model/base/__init__.py,sha256=wmksFIT_909P3YMRKwMzvOVtUHaKQgEVH9XkZ4vJY14,735
74
+ nshtrainer/config/model/config/__init__.py,sha256=Aubl0ypBxiNA67qg6833XBhvCN2EKHkjRv87B4Chq3o,1531
75
+ nshtrainer/config/model/mixins/logger/__init__.py,sha256=shkRA1JaWq4GVCO1DkqTq4v8m7e4o2THyFhfGmAobJQ,550
76
+ nshtrainer/config/nn/__init__.py,sha256=WGKJ4Y5sPWv9VsWeYsgfepIJFvfHp25WXYJEkQ19Ldg,3990
77
+ nshtrainer/config/nn/mlp/__init__.py,sha256=C6dft6w162_J3YggaC5JmpnEjPUcbiw4lSu9FFB0SPU,934
78
+ nshtrainer/config/nn/nonlinearity/__init__.py,sha256=PLYSXOdzd3yub56QQbiQukr3eTagdrgyLIaVtKAEwe8,4797
79
+ nshtrainer/config/optimizer/__init__.py,sha256=F-x3t2IsG0gOMzu226LvDhINiXBTxKw8CiOuUtIzMVs,936
80
+ nshtrainer/config/profiler/__init__.py,sha256=Df7w-VGwDBOYJDYPL9U_5OwwXOjrqQkYb3QFb6o8yyg,1521
81
+ nshtrainer/config/profiler/_base/__init__.py,sha256=smDYCplrI5B38XJcNZ462ZeTo9l1ieLXCxRxf-FJEUY,602
82
+ nshtrainer/config/profiler/advanced/__init__.py,sha256=VCCvbzhEeOcdZ0Unvk_anAcmbQuGogTQhK_bXs5RG9U,892
83
+ nshtrainer/config/profiler/pytorch/__init__.py,sha256=I6mQfAvu1J6bTVci3o3sK8TdOFDBW1cZ1JTWXN8M7Xg,884
84
+ nshtrainer/config/profiler/simple/__init__.py,sha256=jZEke3iaciGrl4k81hGOWMXFZ1CAI9mYAZsvexy8o-c,859
85
+ nshtrainer/config/runner/__init__.py,sha256=rvwxfAHjrYRxhV-LoMi6uN0HtHinAts3qeofeLcQcx4,524
86
+ nshtrainer/config/trainer/_config/__init__.py,sha256=2WDcrUabrlh_dfvp6BDA4_OBmpXSNb-MskER222AQsk,6634
87
+ nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=pSu79zOFFWvqjI3SkHWl13H8ZNJFTc6a5J1r2KnfUKM,667
88
+ nshtrainer/config/util/_environment_info/__init__.py,sha256=mSzPR3mfvMx0ZQrnn2DDyNuCTc2inLKYmHJfEZfsoNk,3730
89
+ nshtrainer/config/util/config/__init__.py,sha256=L-H6GvRWkHZRFnfD-mlYi5cu3Hbtrq4Ih6w7T0TZ5EU,1162
90
+ nshtrainer/config/util/config/dtype/__init__.py,sha256=NCXMVO-EUz3JvPmlDci72O9ZeqpCANs0ieeoXAu9Ljw,550
91
+ nshtrainer/config/util/config/duration/__init__.py,sha256=8llT1MCKQpsdNldN5h5Wo0GjUuRn28Sxw2FTXTNKBpM,1060
92
+ nshtrainer/data/__init__.py,sha256=K4i3Tw4g9EOK2zlMMbidi99y0SyI4w8P7_XUf1n42Ts,260
93
+ nshtrainer/data/balanced_batch_sampler.py,sha256=r1cBKRXKHD8E1Ax6tj-FUbE-z1qpbO58mQ9VrK9uLnc,5481
94
+ nshtrainer/data/datamodule.py,sha256=roT4gfadPpvD-b63_IbzuxmlOjOQI-g3nZJpPvuQXZ0,149
95
+ nshtrainer/data/transform.py,sha256=qd0lIocO59Fk_m90xyOHgFezbymd1mRwly8nbYIfHGc,2263
96
+ nshtrainer/ll/__init__.py,sha256=eSRSqJLHbm7SFaVF5-b-wTwm3oTriRAhpo_ChhwamlM,2513
97
+ nshtrainer/ll/_experimental.py,sha256=lhaCxbUEoZXBn7yd_K0vPch6dg1_TVvSB458tVge2g4,89
98
+ nshtrainer/ll/actsave.py,sha256=zNj16Xv_-holuVI14csp2a-YZjJlONXJIWPnbNsi1VU,245
99
+ nshtrainer/ll/callbacks.py,sha256=WmS0qATlKJYTq5AiY5OY6axmhC5FpV_rtW69wHNcd1k,85
100
+ nshtrainer/ll/config.py,sha256=LFdF3X69loEPQexZ7zkUHRjqyrXpNtaEnerEEn71r-E,181
101
+ nshtrainer/ll/data.py,sha256=TXawppvLXy_N1-CyheHvPuZp_2ZMJQKrmKGgHofNrw4,80
102
+ nshtrainer/ll/log.py,sha256=STn2bIhMOKTQuuS0Jvv01DQhQYpdV4_gQIqjcSNLRts,176
103
+ nshtrainer/ll/lr_scheduler.py,sha256=dis2zS_BqlWn6A-LAkxLbGj1OLjP5OEs9kIEywKXXCE,88
104
+ nshtrainer/ll/model.py,sha256=usuP2kVczhOdGhTxA4_t6fhwrXl-wTslbqh96StprUY,1032
105
+ nshtrainer/ll/nn.py,sha256=G2JjuGSJ_dtK-tQUvtwvMPgwQx7QQ7pA32miksY33iY,78
106
+ nshtrainer/ll/optimizer.py,sha256=m5D3btw3RNGRY37NMrI4yKyglmGBkjnAE37u078keQo,85
107
+ nshtrainer/ll/runner.py,sha256=s424AeQ25NweAxIHyLkSb7CynhhlyQBeZs4rvPK1_pE,153
108
+ nshtrainer/ll/snapshot.py,sha256=IMGBd0xrOppZPXdcgVe2rXBfMSU6QYgtApFM8tVyTqM,123
109
+ nshtrainer/ll/snoop.py,sha256=zPkIoSsZJElG0kWmXnk_YFyQDo1EVH31NcL5wRLxub8,72
110
+ nshtrainer/ll/trainer.py,sha256=iZ-fAsNWAvFwaydXJZrx3_hsQYq10TgZHIJ4sHVfMIo,83
111
+ nshtrainer/ll/typecheck.py,sha256=i8SgaP7C-cWELMWqXGS-7vVegPFUgAF1nj0H_xCV4us,98
112
+ nshtrainer/ll/util.py,sha256=dsdhTO3tGg5OD7NUlmIUyuUbiMupdSUnZfohr3zKb1M,80
113
+ nshtrainer/loggers/__init__.py,sha256=d8aB7NEc57vOFi1Cl7S_Adwk77CenAfT6GbpjunTlDM,484
114
+ nshtrainer/loggers/_base.py,sha256=eTeLKKOl8s_vOhU4_JNwcWEciabDi-elAV7K7PMUOyY,741
115
+ nshtrainer/loggers/csv.py,sha256=ahbQj0uquf6WiIPZV1hUPxz8FuQwvfPww9BHyBvzkJo,1088
116
+ nshtrainer/loggers/tensorboard.py,sha256=yP9hebra1NeNiEAB7WrlJzh1AGYHLOvC0vP_Ku82sfE,2252
117
+ nshtrainer/loggers/wandb.py,sha256=VTPy1HtAgnzmvTYYuHJ7cL1exOL95nyd93jlm65OY5o,6746
118
+ nshtrainer/lr_scheduler/__init__.py,sha256=o7eDaj-KOetQvEwMjGKpcQofs1jimqUzHbJ5YM5cOMY,772
119
+ nshtrainer/lr_scheduler/_base.py,sha256=WnD7_fvkXckMbIc-ZkcbLm9umf3vm1NRJLWA5fiC8Bk,3695
120
+ nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=g9F5EXXW4LRoOXP6WY3V-5cYnDLjlRKkJdh7YrOrWnc,5406
121
+ nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=phDJgjVGFUGurWFIECcKOxK-EdBeTcDxIlcSl9X-988,2818
122
+ nshtrainer/metrics/__init__.py,sha256=Nqkn_jsDf3n5WtfMcnaaEftYjIIT2b-S7rmsB1MOMkU,86
123
+ nshtrainer/metrics/_config.py,sha256=kR9OJLZXIaZpG8UpHG3EQwOL36HB8yPk6afYjoIM0XM,1324
124
+ nshtrainer/model/__init__.py,sha256=bKgE221066M2QlmLMjMCyh0qhjpMWWaWTAfW9HLKgAE,297
125
+ nshtrainer/model/base.py,sha256=lUYrkbBYdbTmOr94HnvS_ZLQlaWRWxEH3URdQ00sqFs,18645
126
+ nshtrainer/model/config.py,sha256=ez1XgWdyZTxh6ETWE51p7CEH5o_owVgSIqUlYXEv56M,6997
127
+ nshtrainer/model/mixins/callback.py,sha256=USsff85bvcrOOZPyYbs6AxfcqkO7RyCbzEfbp5nFANs,2455
128
+ nshtrainer/model/mixins/logger.py,sha256=b6aH9A9SeQOetBNapVVfD3wCrJgSKWJu6s8biwazbBc,5327
129
+ nshtrainer/nn/__init__.py,sha256=sANhrZpeN5syLKOsmXMwhaFl2SBFPWcLaEe1EH22TWQ,1463
130
+ nshtrainer/nn/mlp.py,sha256=2W8bzE96DzCMzGm6WPiPhNFQfhqaoG3GXPn_oKBnlUM,5988
131
+ nshtrainer/nn/module_dict.py,sha256=9plb8aQUx5TUEPhX5jI9u8LrpTeKe7jZAHi8iIqcN8w,2365
132
+ nshtrainer/nn/module_list.py,sha256=UB43pcwD_3nUke_DyLQt-iXKhWdKM6Zjm84lRC1hPYA,1755
133
+ nshtrainer/nn/nonlinearity.py,sha256=mp5XvXRHURB6jwuZ0YyTj5ZoHJYNJNgO2aLtUY1D-2Y,6114
134
+ nshtrainer/optimizer.py,sha256=73UWfuTU4moDQNSKI8VSQSDc6_i7mBtemxvhBkM-fC4,1563
135
+ nshtrainer/profiler/__init__.py,sha256=X6E4cbIRaXHCDgQtT3LFPw5xIRBhqnJ03SVFbsD-Yd8,516
136
+ nshtrainer/profiler/_base.py,sha256=uPlH3BIBMK22upQQe2cLmp_jl20WV1P9SaEg2s_sEy0,933
137
+ nshtrainer/profiler/advanced.py,sha256=I1pJckyt4sF_y-KAnoCC1cfZMBQyOj3v-Hv13m175kE,1148
138
+ nshtrainer/profiler/pytorch.py,sha256=mILwyMTPpqMkHxiijd5GTatw50mo435ZMYfbuDcNKCs,2745
139
+ nshtrainer/profiler/simple.py,sha256=Z86mhUbQ04RlgHo6EhJ-9PcROebjVObk6ybK-zEQ4QY,993
140
+ nshtrainer/runner.py,sha256=UIaTPhgmkM_1oF0Qv_av9VMr8DgJJ4OFASz3n8VT19E,3189
141
+ nshtrainer/scripts/find_packages.py,sha256=-oNnSNPp3pujCVgManW_LFlJcnvhrtHgvUJ4W88-6-o,1460
142
+ nshtrainer/trainer/__init__.py,sha256=tvb6xb3lMfXGagnDcUKT0k-QZQFshUWdpErFGZOjQfo,76
143
+ nshtrainer/trainer/_config.py,sha256=vexsMRDbGuHFUFbNASw-t5lrcq7qM506Dy-6Zwska3k,29399
144
+ nshtrainer/trainer/_runtime_callback.py,sha256=T3epaj1YeIN0R8CS2cg5HNJIB21TyaD_PVNNOPJ6nJs,4200
145
+ nshtrainer/trainer/checkpoint_connector.py,sha256=6ekND0oiIx7nVQBBHSlI_pqmtGifrCcnIe_6Z23AyCo,2383
146
+ nshtrainer/trainer/signal_connector.py,sha256=YMJf6vTnW0JcnBkuYikm9x_9XscaokrCEzCn4THOGao,10776
147
+ nshtrainer/trainer/trainer.py,sha256=RLeFU0sP_-NFWJ62GYOQuGP2369Oetj52e5yfaQsi2M,16875
148
+ nshtrainer/util/_environment_info.py,sha256=B6IABVgmZVEp2x-zhf0eqjC9_CUuJoqyyzIH44yGDuw,24249
149
+ nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
150
+ nshtrainer/util/bf16.py,sha256=9QhHZCkYSfYpIcxwAMoXyuh2yTSHBzT-EdLQB297jEs,762
151
+ nshtrainer/util/config/__init__.py,sha256=Z39JJufSb61Lhn2GfVcv3eFW_eorOrN9-9llDWlnZZM,272
152
+ nshtrainer/util/config/dtype.py,sha256=LBndiDd00c29ULFrmG1R7yPvPO4fxHqFYQGacAx8BPA,2001
153
+ nshtrainer/util/config/duration.py,sha256=mM-UfU_HvhXwW33TYEDg0x58n80tnle2e6VaWtxZTjk,764
154
+ nshtrainer/util/environment.py,sha256=s-B5nY0cKYXdFMdNYumvC_xxacMATiI4DvV2gUDu20k,4195
155
+ nshtrainer/util/path.py,sha256=mfuMRk29YpmwdRsyyf99VT0n_T9caM99SS7VfA6oIHE,3501
156
+ nshtrainer/util/seed.py,sha256=diMV8iwBKN7Xxt5pELmui-gyqyT80_CZzomrWhNss0k,316
157
+ nshtrainer/util/slurm.py,sha256=HflkP5iI_r4UHMyPjw9R4dD5AHsJUpcfJw5PLvGYBRM,1603
158
+ nshtrainer/util/typed.py,sha256=Xt5fUU6zwLKSTLUdenovnKK0N8qUq89Kddz2_XeykVQ,164
159
+ nshtrainer/util/typing_utils.py,sha256=5EDJ2C_DEZauWwWOyNWBCVOJJL3znHPkayJcRq3SHPk,421
160
+ nshtrainer-0.43.0.dist-info/METADATA,sha256=z2RdjeMxEuaeRNNJW_Dv27YJ2WwzV4WWYn0KEyBpAks,916
161
+ nshtrainer-0.43.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
162
+ nshtrainer-0.43.0.dist-info/RECORD,,
@@ -1,162 +0,0 @@
1
- nshtrainer/__init__.py,sha256=8hx3uqzroMYOQM7S4U3Eznw7MHF3YDqUkO1MDQeC4cM,642
2
- nshtrainer/_callback.py,sha256=A1zLsTy4b_wOYnInLLXGSRdHzT2yNa6mPEql-ozm0u0,1013
3
- nshtrainer/_checkpoint/loader.py,sha256=5vjg-OFChXJjgiOVv8vnV8nwTscfdDtEdxQRz6uPfDE,14158
4
- nshtrainer/_checkpoint/metadata.py,sha256=5D4PgKodzhLsmQvuF3xxkH49epKaegxi4wh_ImDTtns,4737
5
- nshtrainer/_checkpoint/saver.py,sha256=xuA9iDpc4wAs4ieYW1eJoJicMn6e5xCeCIEKfVOCvrY,1324
6
- nshtrainer/_directory.py,sha256=RjnW6vKTeKlz2vQWT3cG0Jje5BkFXA7HpUubDhcSiq4,2993
7
- nshtrainer/_experimental/__init__.py,sha256=pEXPyI184UuDHvfh4p9Kg9nQZQZI41e4_HvNd4BK-yg,81
8
- nshtrainer/_hf_hub.py,sha256=woRRy9bzAQ8ueoH903CzYGjQYZ0Cp7Hw5uv8qgLnZ5o,14105
9
- nshtrainer/callbacks/__init__.py,sha256=y4QKDiI4IykcHncDOE-OemFFQqegxyeoD_3v9i2OwFw,3248
10
- nshtrainer/callbacks/_throughput_monitor_callback.py,sha256=aJo_11rc4lo0IYOd-kHmPDtzdC4ctgXyRudkRJqH4m4,23184
11
- nshtrainer/callbacks/actsave.py,sha256=qbnaKts4_dvjPeAaPtv7Ds12_vEWzaHUfg_--49NB9I,4041
12
- nshtrainer/callbacks/base.py,sha256=NpjeKmonJ1Kaz5_39XSn3LlDwvbGjk6WV8BpHSNCvI4,3508
13
- nshtrainer/callbacks/checkpoint/__init__.py,sha256=g-3zIthupERKqWZQw-A_busQPaPRkto6iHBV-M7nK1Y,527
14
- nshtrainer/callbacks/checkpoint/_base.py,sha256=X6844ceZ-EMz9UCVLJnA1d-ej2AkkRH90v2hy2zPDkc,6215
15
- nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=MOC5jLlQnnc61yuiVc2-O4NeYclnkXroOPDV9-2zh5w,2553
16
- nshtrainer/callbacks/checkpoint/last_checkpoint.py,sha256=CWWv0cSwQ1VAX26N7hAyMxbNCk26Keh39oQguBEK5To,1102
17
- nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=ctT88EGT22_t_6tr5r7Sfo43cuve6XeroBnBYRMPOus,3372
18
- nshtrainer/callbacks/debug_flag.py,sha256=T7rkY9hYQ_-PsPo2XiQ4eVZ9bBsTd2knpZWctCbjxXc,2011
19
- nshtrainer/callbacks/directory_setup.py,sha256=c0uY0oTqLcQ3egInHO7G6BeQQgk_xvOLoHH8FR-9U0U,2629
20
- nshtrainer/callbacks/early_stopping.py,sha256=upCNtjkXmwGPwLJmDZacXdBzySG2i_sGIK9QjnaE6tU,4638
21
- nshtrainer/callbacks/ema.py,sha256=8-WHmKFP3VfnzMviJaIFmVD9xHPqIPmq9NRF5xdu3c8,12131
22
- nshtrainer/callbacks/finite_checks.py,sha256=gJC_RUr3ais3FJI0uB6wUZnDdE3WRwCix3ppA3PwQXA,2077
23
- nshtrainer/callbacks/gradient_skipping.py,sha256=EBNkANDnD3BTszWjnG-jwY8FEj-iRqhE3e1x5LQF6M8,3393
24
- nshtrainer/callbacks/interval.py,sha256=smz5Zl8cN6X6yHKVsMRS2e3SEkzRCP3LvwE1ONvLfaw,8080
25
- nshtrainer/callbacks/log_epoch.py,sha256=fTa_K_Y8A7g09630cG4YkDE6AzSMPkjb9bpPm4gtqos,1120
26
- nshtrainer/callbacks/norm_logging.py,sha256=T2psu8mYsw9iahPKT6aUPjkGrZ4TIzm6_UUUmE09GJs,6274
27
- nshtrainer/callbacks/print_table.py,sha256=_FdAHhqylWGk4Z0c2FrLFeiMA4jhfA_beZRK_BHpzmE,2837
28
- nshtrainer/callbacks/rlp_sanity_checks.py,sha256=c30G9jAu42QLLIS5LnusdSnI3wqyIHgOUFDRcKESuNI,9935
29
- nshtrainer/callbacks/shared_parameters.py,sha256=fqlDweFDXPV_bfcAWpRgaJIad9i5AehYDtuJjDtUum4,2922
30
- nshtrainer/callbacks/throughput_monitor.py,sha256=H_ocXErZxUO3dxFk8Tx_VQdpI9E_Ztvqof5WtFevLyQ,1838
31
- nshtrainer/callbacks/timer.py,sha256=quS79oYClDUvQxJkNWmDMe0hwRUkkREgTgqzVrnom50,4607
32
- nshtrainer/callbacks/wandb_upload_code.py,sha256=OWG4UkL2SfW6oj6AGRXeBJsZmgsqeHLW2Fj8Jm4ga3I,2298
33
- nshtrainer/callbacks/wandb_watch.py,sha256=Y6SEXfIx3kDDQbI5zpP53BVq0FBLJbLd3RJsiHZk1-Y,2921
34
- nshtrainer/config/__init__.py,sha256=fFp5PCXKVVJqiv4gONdo4fpDNReX8uSQTWliUQmBa60,21848
35
- nshtrainer/config/_checkpoint/loader/__init__.py,sha256=hCjXcva0NMMezDumPYoBxKdXkNv8IhALgGforQjUbWw,2363
36
- nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=oCpMBf7PduAaK6bxcNqdzVNdCWR2kvbOEvbgIKo166M,831
37
- nshtrainer/config/_directory/__init__.py,sha256=I5CO98ktOWzCGmtcYlWYt3Fa0BwK4MnxxnKaNuVet_c,914
38
- nshtrainer/config/_hf_hub/__init__.py,sha256=qNUK71CnviYcvd86Jjc7ZYFWjr6PLmYCynsVE7QmvQQ,1027
39
- nshtrainer/config/callbacks/__init__.py,sha256=xEemxVCAaQA-iOnpp-ul-c3h_2DGjr4HGfwyavB01Sg,6491
40
- nshtrainer/config/callbacks/actsave/__init__.py,sha256=ySlBq2LJbJgSsPtpT-raEiGSnZ3oFGTFB-MOCfZQlTM,773
41
- nshtrainer/config/callbacks/base/__init__.py,sha256=HWeQLSg8wGFh9pFgw0eyRkfW2uO3KzM1Wgf-s6kmcQo,566
42
- nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=m1Xfmi16feKWry_dvz8oAS5Zlj-tT96EpY47eNmVMTQ,2664
43
- nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=J3uc2rnMF2ZDioSIi3J5UrGRE52KA5zKoU6bR6M__TI,1219
44
- nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=Ypgp3zt1W30zOvwGsCkrwq15KLe0ujlnJ_-MDlhM6i8,1601
45
- nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=fL4sfS6BjIHBa4Kaaqmex6HSKw_y_maad78uzYiTVb0,1319
46
- nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=L0cXPoWyZtOyo9ckyeggPuGanN_iCnHOXRpmN_bulxw,1033
47
- nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=Rplsl1OhADFF4v4PK_1tYhlPMzgfH70YAS8gmR_Ev4w,872
48
- nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=H8cckUCNuwmiIr_zYzh1robpRYRi2nNttl5IKaX6D1c,897
49
- nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=aths-c9sGApQEzikMSWLjBICpNz2mtCO8--DsW8ZjiM,1130
50
- nshtrainer/config/callbacks/ema/__init__.py,sha256=xgAgcNBMAusIurhcV_W-95aW-g5jwPmNrtW9FwsW4YU,741
51
- nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=Ts4eCT26kuwQj0jE0VjZARuQjMB0hATE-KutM6PRmns,881
52
- nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=Hxk-feSH8H1T6nuTvSMt4yDOiSQ_h8Ob3dNlj3n6GM4,913
53
- nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=qnBnh1PTbomp9CpiniHwZs5BIW8Yi1YmLL-TDOA58Mk,856
54
- nshtrainer/config/callbacks/print_table/__init__.py,sha256=rQDfNJHVpBNlG2en3ZQ-hCTN4iPl-7jIGi6owKOfkSc,893
55
- nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256=0pb2FeArOZAQ0tNxjg1QwemsfoSim-jLDXFv0vlHGJI,909
56
- nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=27EcGdNahNCWnu7pM62xelnfu4WMosrWrsudsNnVTjk,913
57
- nshtrainer/config/callbacks/throughput_monitor/__init__.py,sha256=eui6qangrb3errf2jJ7slcY04aJ2sOBt8wyNDODeOlA,921
58
- nshtrainer/config/callbacks/timer/__init__.py,sha256=3P-RroY0QGC_ayNbcQ7HqOUL8l7UcHIe9OJ416ve-h4,807
59
- nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=XOFTN8ua7xHpXN63G1ZirqKTdGUs6i1yuwALFunxK-4,909
60
- nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=TFykMCR_r40vyEL7enne96aE4Zy3YWaPcntpS26Giww,848
61
- nshtrainer/config/loggers/__init__.py,sha256=R8veYxtV-dO-AeLCQ48oemUIxA3iNLw0OjpWafn9Dxo,2126
62
- nshtrainer/config/loggers/_base/__init__.py,sha256=tD6WYqYmZ0qofaB7_XLIV347hwfd1XyrskAMVGyk0wA,526
63
- nshtrainer/config/loggers/csv/__init__.py,sha256=sUTZG1pMNtaRzRLicGP7cYD-mWbfOWvhz492VOm0A9M,719
64
- nshtrainer/config/loggers/tensorboard/__init__.py,sha256=O2WyxdWnQuKZLu-KooURC6k9Z4EAbplcjQc22boZY6k,860
65
- nshtrainer/config/loggers/wandb/__init__.py,sha256=TGn6s1ofUEnO_jFBn8mdcjtDNc0avwDo1lJPbGUX6h0,1438
66
- nshtrainer/config/lr_scheduler/__init__.py,sha256=fMUIOn7r4iIThhvqL0xGE-J5GVgbXQWUeprfpiYRHQY,2113
67
- nshtrainer/config/lr_scheduler/_base/__init__.py,sha256=W68UxqYwSUHybd6rt7p2F0stNImbjUNN_aV8CDcXGhg,603
68
- nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py,sha256=x1dl3U7YajRWNOkBmR8n4Q-P2dCrHTQ3tcTpfVi7jZg,1305
69
- nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py,sha256=sRYCkXVcB0xQ3rOppigkNb7Hcz8-so_oLT8TUn8WBu0,1229
70
- nshtrainer/config/metrics/__init__.py,sha256=MD3ltGcrXdYZmvllVGj1yoU3UOHsB-z46QGugEVDXiE,532
71
- nshtrainer/config/metrics/_config/__init__.py,sha256=AJ-by6EzIkmBAxEzV5vgRUmVt-rsw2muPJBbkw8-BhY,514
72
- nshtrainer/config/model/__init__.py,sha256=z0m6aZIUrx1B399-VpG_B-rWdJdqWXg7GlQ5SQiZaQw,1525
73
- nshtrainer/config/model/base/__init__.py,sha256=Uot7t00zZpj4hZNay1AyvgKeClDYktRZqVyjnMmrxd8,699
74
- nshtrainer/config/model/config/__init__.py,sha256=OIyHPEZtzxcKEZorchrBTxCXgvLAFKUraJW21JeF0mc,1495
75
- nshtrainer/config/model/mixins/logger/__init__.py,sha256=kBWS3jdySOv77p7QuZtWMlw7rWdFWqodTeU6L8T9Q4Y,514
76
- nshtrainer/config/nn/__init__.py,sha256=SW9ZIoqlXGRHEromIgUwlLlaAyk3thf0GPSTZvXGCtM,3954
77
- nshtrainer/config/nn/mlp/__init__.py,sha256=Kqjklbj9ny7vGgf7EC4WA226JVauP5vdGK_orQzv4uk,898
78
- nshtrainer/config/nn/nonlinearity/__init__.py,sha256=BIyxKR3S-1Fvf2JfwfkgZWu-6hr4nvjCa8ugHtjcQ0Q,4761
79
- nshtrainer/config/optimizer/__init__.py,sha256=gOxCvg8Cl7FFSKZEekcIKRZMbpBZgXMiUFxOXrHWO4U,900
80
- nshtrainer/config/profiler/__init__.py,sha256=OQb_MdbGni-hFHip_NrctnoIeG6Cks_8b2k3nPQh5XU,1485
81
- nshtrainer/config/profiler/_base/__init__.py,sha256=zJ1dQIroQMmMkDYq-hAfLCahpAKAmfzI6eKf-SCzV2E,566
82
- nshtrainer/config/profiler/advanced/__init__.py,sha256=53jR1agoCilDW8bKuT0NPgjQRwFy-GU38NFLnGkrpIU,856
83
- nshtrainer/config/profiler/pytorch/__init__.py,sha256=LBxTukfv19-ISLg14r62a4YXERHnZiF1-QECfTtSD8M,848
84
- nshtrainer/config/profiler/simple/__init__.py,sha256=dbRl6LWnr6aoIEf3sW_vwM1LAz2Nn3jrO0rltlayk4s,823
85
- nshtrainer/config/runner/__init__.py,sha256=U5JWcg1u2oErwRnLwDiyArxDFutLWuordpDSaAJPtRs,488
86
- nshtrainer/config/trainer/_config/__init__.py,sha256=7ZokJsepbCwP_rXOmvdEDxLYWhVkU8WnXQD4aJ5G6gs,6485
87
- nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=zR6sS2bZiaWnpQqixXQ9ki7qbMLxVqXozZpjXThcGjw,631
88
- nshtrainer/config/util/_environment_info/__init__.py,sha256=hAX0-OtzahTKXkBURaPorREO4Wb-Ok_gsHHpFZ3p-GI,3694
89
- nshtrainer/config/util/config/__init__.py,sha256=rQ_YrelDqwv9eYwHafgyZQOLA0fSV4EH7png-IkecdQ,1126
90
- nshtrainer/config/util/config/dtype/__init__.py,sha256=Sik13kqmQvFn909ajIe4ISMKQgqn_P7D_Ce5S4o9Pig,514
91
- nshtrainer/config/util/config/duration/__init__.py,sha256=KRAiveUvazReYX2U7JXMKyi8xIq8sbht50J8-Fabg0Q,1024
92
- nshtrainer/data/__init__.py,sha256=fnigdEtNcB8DJqz4MHvcI6UwOZICFzWItskY78lLTMA,224
93
- nshtrainer/data/balanced_batch_sampler.py,sha256=ybMJF-CguaZ17fLEweZ5suaGOiHOMEm3Bn8rQfGTzGQ,5445
94
- nshtrainer/data/datamodule.py,sha256=Dk06LdcbK13sCLQWacdCoXU_CHWZxoKEZ_xpAg7nZQg,113
95
- nshtrainer/data/transform.py,sha256=6SNs3_TpNpfhcwTwvPKyEJ3opM1OT7LmMEYQNHKgRl8,2227
96
- nshtrainer/ll/__init__.py,sha256=L-aTi1V1bbvnZjOro8NvI393zbHQSFR9movWSRK9Mds,2477
97
- nshtrainer/ll/_experimental.py,sha256=oBQCKOEVYoxuUU9eLb-Fg2B2mzZD7SA0zfAO6lmWZ88,53
98
- nshtrainer/ll/actsave.py,sha256=2lbiseSrjcwFT6AiyLNWarTWl1bnzliVWlu1iOfnP30,209
99
- nshtrainer/ll/callbacks.py,sha256=AxyUmc8aGRSjx6WwwgXYCmdJ73rwLuEAEH0AGRosojQ,49
100
- nshtrainer/ll/config.py,sha256=fKumJf42HY2FITX1QUM1OTXkYD6U2np2ciyd4PFRPZ8,145
101
- nshtrainer/ll/data.py,sha256=zRG0FRje-jtSHximVzkHIHzpwsyQxpHCoACFihNKLPM,44
102
- nshtrainer/ll/log.py,sha256=d4BB3TyM8imK65EXOiOeUTF0zFM1ropbe7Vq3DeB0xU,140
103
- nshtrainer/ll/lr_scheduler.py,sha256=7xjhN6L69BCUzFhcy33NtMtPuCzHiB611zVWFg92lQ0,52
104
- nshtrainer/ll/model.py,sha256=Cw8Vq8IUL6YU1fTUcOIZsXcNJ3XyKgQY4YENIsL9H7c,996
105
- nshtrainer/ll/nn.py,sha256=8qiRDFwojIxkB7-LtNWk4mLL2tJbaskHYofDsOIHiNg,42
106
- nshtrainer/ll/optimizer.py,sha256=3T-VZtT73jVvwCNJGDjgGEbzs-1LFTzMQH-SB_58mSo,49
107
- nshtrainer/ll/runner.py,sha256=B0m5VEhNKIjF1aFmqPkonkQxDoRL2jeHZGsV3zwhSVE,117
108
- nshtrainer/ll/snapshot.py,sha256=EdzvbE6IyHcNpIvkF1lWj3pBigfRF9jQXgOlbgBavYs,87
109
- nshtrainer/ll/snoop.py,sha256=hG9VCdm8mIZytHLZgUKRSoWqg55rVvUpBAH-OSiwgvI,36
110
- nshtrainer/ll/trainer.py,sha256=hkn2xPtrSPQ7LqQhbyAKuMfNyHdhqB9bDPvgRCK8oJM,47
111
- nshtrainer/ll/typecheck.py,sha256=ryV1Tzcf7hJ4I19H1oQVkikU9spmRk8jyIKQZ5UF7pQ,62
112
- nshtrainer/ll/util.py,sha256=PQGu5Ff1raizxjXdm2rFFu4Mo816dBIYmItjkJX1_qk,44
113
- nshtrainer/loggers/__init__.py,sha256=C_xk0A3_qKbNdTmzK85AgjRHFD3w-jPRS2ig-iPhfEk,448
114
- nshtrainer/loggers/_base.py,sha256=xiZKEK0ALJkcqf4OpVNRY0QbZsamR_WR7x7m_68YHXQ,705
115
- nshtrainer/loggers/csv.py,sha256=D_lYyd94bZ8jAgnRo-ARtFgVcInaD9zktxtsUD9RWCI,1052
116
- nshtrainer/loggers/tensorboard.py,sha256=wL2amRSdP68zbslZvBeM0ZQBnjF3hIKsz-_lBbdomaM,2216
117
- nshtrainer/loggers/wandb.py,sha256=GMlEzkFesgS3qgVgYqATPHJ3rvs9etwFlk3HbidOyRQ,6662
118
- nshtrainer/lr_scheduler/__init__.py,sha256=uEvgaFAs-4s_bAEMaildy0GT6OvgpgOEKTuzqutESHE,736
119
- nshtrainer/lr_scheduler/_base.py,sha256=7xOIuxQ86YHbFWG5a3gX46emQj1WN_LaY4-i0Q1TDBg,3659
120
- nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=E8LW78uuby7bIsoLPpcF1bmNK4lSko-r3qPL-vuHWXQ,5370
121
- nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=h76oTHYpMxauV_l6lviya5DW-WKArwxxf7ZQizhmbCw,2782
122
- nshtrainer/metrics/__init__.py,sha256=ObLIELGguIEcUpRsUkqh1ltrvZii6vglTpJGrPvoy00,50
123
- nshtrainer/metrics/_config.py,sha256=jgRBfDAQLFTW7AiUY7CRtdfts6CR6keeuqm0FFMWCzQ,1288
124
- nshtrainer/model/__init__.py,sha256=2i_VEy6u_Y1LUGKljHXWeekvhnUcanZM2QyaaBM1Bmw,261
125
- nshtrainer/model/base.py,sha256=NasbYZJBuEly6Hm9t9HVZk-CUHmy4T7p1v-Ye981XA4,18609
126
- nshtrainer/model/config.py,sha256=gyP5uuDfl4EbkmFlqnpNm49D0PgJWK9S6UOFGGaAAtQ,6961
127
- nshtrainer/model/mixins/callback.py,sha256=rbe8P22iEjPkH1df6rfEo3Txw7EwSz6Dkm0TWO_AysM,2419
128
- nshtrainer/model/mixins/logger.py,sha256=xOymSTofukEYZGkGojXsMEO__ZlBI5lIPZVmlotMEX8,5291
129
- nshtrainer/nn/__init__.py,sha256=0QPFl02a71WZQjLMGOlFNMmsYP5aa1q3eABHmnWH58Q,1427
130
- nshtrainer/nn/mlp.py,sha256=V0FrScpIUdg_IgIO8GMtIsGEtmHjwF14i2IWxmZrsqg,5952
131
- nshtrainer/nn/module_dict.py,sha256=NOY0B6WDTnktyWH4GthsprMQo0bpehC-hCq9SfD8paE,2329
132
- nshtrainer/nn/module_list.py,sha256=bbR62bnQM4No6HtmBllijOhLwDtWwmcPcO2DklPbsbc,1719
133
- nshtrainer/nn/nonlinearity.py,sha256=4sYE4MN5zojc-go1k0PYtqssVRuXrM7D4tbpIXp5K-E,6078
134
- nshtrainer/optimizer.py,sha256=kuJEA1pvB3y1FcsfhAoOJujVqEZqFHlmYO8GW6JeA1g,1527
135
- nshtrainer/profiler/__init__.py,sha256=cJ_wAm8j3Bz6cKgNQ_9gQLZz9nddKW53VE81UKg8l8g,480
136
- nshtrainer/profiler/_base.py,sha256=YF5lsJBIl9qts9GLW5Z62JuYeo4SnIArhlFwTGkfTb4,897
137
- nshtrainer/profiler/advanced.py,sha256=44asloha0aGUW8YwjQt3lm3ve8H-N6mM4QgseUSLT30,1112
138
- nshtrainer/profiler/pytorch.py,sha256=tGeRvoPP5ulWX2RkfXrQvMBoki1T95dpz5p8mwyon1I,2709
139
- nshtrainer/profiler/simple.py,sha256=MbMfsJvligd0mtGiltxJ0T8MQVDP9T9BzQZFwswl66Y,957
140
- nshtrainer/runner.py,sha256=llW5PThXfzlu8ym1waFVzED1RGI91OHqN3j02sepoZo,3153
141
- nshtrainer/scripts/find_packages.py,sha256=ixYivZobumyyGsf2B9oYMLyLTRcBzY_vUv-u3bNW-hs,1424
142
- nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
143
- nshtrainer/trainer/_config.py,sha256=YqpGb4RodkUg87TVE5WBSc4CQkUF0z3qDRdil1HRxoM,29198
144
- nshtrainer/trainer/_runtime_callback.py,sha256=sd2cUdRJG-UCdQr9ruZvEYpNGNF1t2W2fuxwwVlQD9E,4164
145
- nshtrainer/trainer/checkpoint_connector.py,sha256=r0ir4xYSdf_jebM0x09qaO6nJsvsiRQDyM0fs80ppOQ,2347
146
- nshtrainer/trainer/signal_connector.py,sha256=2EzkVktlasl8PgWAKNLDZRUMY__gRlDy1HdinAU-tfU,10740
147
- nshtrainer/trainer/trainer.py,sha256=8T4LB31ygXXS3DECkvD2uqgElAxkulacYvZyL_-imJs,16839
148
- nshtrainer/util/_environment_info.py,sha256=CFUUZYjXhBLWGc0jtPNOaZgYMueUDEHpEaWFA1f3GoY,24213
149
- nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
150
- nshtrainer/util/bf16.py,sha256=VUnIG6aA4XtZscZc_dxv5ln_jlEbdU3eMFwDb5SEWSI,726
151
- nshtrainer/util/config/__init__.py,sha256=o8fwPf_dctE_7CAkT0wNOBkvmxnzYzXeHpLedrZLt54,236
152
- nshtrainer/util/config/dtype.py,sha256=JtYjrcBFNBlziJnLAE6QS0QV4PUXhGspYH1hNFrB3ks,1965
153
- nshtrainer/util/config/duration.py,sha256=pgIKQ88Dg8y1YAKUvUsNWu9hc9O79kdYBfgmC3a_-kQ,728
154
- nshtrainer/util/environment.py,sha256=AeW_kLl-N70wmb6L_JLz1wRj0kA70xs6RCmc9iUqczE,4159
155
- nshtrainer/util/path.py,sha256=BmBzqtxGu7w2gTYkxQevdBffu_Crm1dH3TURoqqjR6M,3465
156
- nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
157
- nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
158
- nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
159
- nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
160
- nshtrainer-0.42.0.dist-info/METADATA,sha256=KOHvIES5urRR9kMl8VrBaSNbz40XAl2cIpTaDMj08E8,916
161
- nshtrainer-0.42.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
162
- nshtrainer-0.42.0.dist-info/RECORD,,