nshtrainer 0.44.1__py3-none-any.whl → 1.0.0b10__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 (124) hide show
  1. nshtrainer/__init__.py +6 -3
  2. nshtrainer/_callback.py +297 -2
  3. nshtrainer/_checkpoint/loader.py +23 -30
  4. nshtrainer/_checkpoint/metadata.py +22 -18
  5. nshtrainer/_experimental/__init__.py +0 -2
  6. nshtrainer/_hf_hub.py +25 -26
  7. nshtrainer/callbacks/__init__.py +1 -3
  8. nshtrainer/callbacks/actsave.py +22 -20
  9. nshtrainer/callbacks/base.py +7 -7
  10. nshtrainer/callbacks/checkpoint/__init__.py +1 -1
  11. nshtrainer/callbacks/checkpoint/_base.py +8 -5
  12. nshtrainer/callbacks/checkpoint/best_checkpoint.py +4 -4
  13. nshtrainer/callbacks/checkpoint/last_checkpoint.py +1 -1
  14. nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py +4 -4
  15. nshtrainer/callbacks/debug_flag.py +14 -19
  16. nshtrainer/callbacks/directory_setup.py +6 -11
  17. nshtrainer/callbacks/early_stopping.py +3 -3
  18. nshtrainer/callbacks/ema.py +1 -1
  19. nshtrainer/callbacks/finite_checks.py +1 -1
  20. nshtrainer/callbacks/gradient_skipping.py +1 -1
  21. nshtrainer/callbacks/log_epoch.py +1 -1
  22. nshtrainer/callbacks/norm_logging.py +1 -1
  23. nshtrainer/callbacks/print_table.py +1 -1
  24. nshtrainer/callbacks/rlp_sanity_checks.py +1 -1
  25. nshtrainer/callbacks/shared_parameters.py +1 -1
  26. nshtrainer/callbacks/timer.py +1 -1
  27. nshtrainer/callbacks/wandb_upload_code.py +1 -1
  28. nshtrainer/callbacks/wandb_watch.py +1 -1
  29. nshtrainer/config/__init__.py +189 -189
  30. nshtrainer/config/_checkpoint/__init__.py +70 -0
  31. nshtrainer/config/_checkpoint/loader/__init__.py +6 -6
  32. nshtrainer/config/_directory/__init__.py +2 -2
  33. nshtrainer/config/_hf_hub/__init__.py +2 -2
  34. nshtrainer/config/callbacks/__init__.py +44 -44
  35. nshtrainer/config/callbacks/checkpoint/__init__.py +11 -11
  36. nshtrainer/config/callbacks/checkpoint/_base/__init__.py +4 -4
  37. nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py +8 -8
  38. nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py +4 -4
  39. nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py +4 -4
  40. nshtrainer/config/callbacks/debug_flag/__init__.py +4 -4
  41. nshtrainer/config/callbacks/directory_setup/__init__.py +4 -4
  42. nshtrainer/config/callbacks/early_stopping/__init__.py +4 -4
  43. nshtrainer/config/callbacks/ema/__init__.py +2 -2
  44. nshtrainer/config/callbacks/finite_checks/__init__.py +4 -4
  45. nshtrainer/config/callbacks/gradient_skipping/__init__.py +4 -4
  46. nshtrainer/config/callbacks/{throughput_monitor → log_epoch}/__init__.py +8 -10
  47. nshtrainer/config/callbacks/norm_logging/__init__.py +4 -4
  48. nshtrainer/config/callbacks/print_table/__init__.py +4 -4
  49. nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py +4 -4
  50. nshtrainer/config/callbacks/shared_parameters/__init__.py +4 -4
  51. nshtrainer/config/callbacks/timer/__init__.py +4 -4
  52. nshtrainer/config/callbacks/wandb_upload_code/__init__.py +4 -4
  53. nshtrainer/config/callbacks/wandb_watch/__init__.py +4 -4
  54. nshtrainer/config/loggers/__init__.py +10 -6
  55. nshtrainer/config/loggers/actsave/__init__.py +29 -0
  56. nshtrainer/config/loggers/csv/__init__.py +2 -2
  57. nshtrainer/config/loggers/wandb/__init__.py +6 -6
  58. nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py +4 -4
  59. nshtrainer/config/nn/__init__.py +18 -18
  60. nshtrainer/config/nn/nonlinearity/__init__.py +26 -26
  61. nshtrainer/config/optimizer/__init__.py +2 -2
  62. nshtrainer/config/profiler/__init__.py +2 -2
  63. nshtrainer/config/profiler/pytorch/__init__.py +4 -4
  64. nshtrainer/config/profiler/simple/__init__.py +4 -4
  65. nshtrainer/config/trainer/__init__.py +180 -0
  66. nshtrainer/config/trainer/_config/__init__.py +59 -36
  67. nshtrainer/config/trainer/trainer/__init__.py +27 -0
  68. nshtrainer/config/util/__init__.py +109 -0
  69. nshtrainer/config/util/_environment_info/__init__.py +20 -20
  70. nshtrainer/config/util/config/__init__.py +2 -2
  71. nshtrainer/data/datamodule.py +52 -2
  72. nshtrainer/loggers/__init__.py +2 -1
  73. nshtrainer/loggers/_base.py +5 -2
  74. nshtrainer/loggers/actsave.py +59 -0
  75. nshtrainer/loggers/csv.py +5 -5
  76. nshtrainer/loggers/tensorboard.py +5 -5
  77. nshtrainer/loggers/wandb.py +17 -16
  78. nshtrainer/lr_scheduler/reduce_lr_on_plateau.py +9 -7
  79. nshtrainer/model/__init__.py +0 -4
  80. nshtrainer/model/base.py +64 -347
  81. nshtrainer/model/mixins/callback.py +24 -5
  82. nshtrainer/model/mixins/debug.py +86 -0
  83. nshtrainer/model/mixins/logger.py +142 -145
  84. nshtrainer/profiler/_base.py +2 -2
  85. nshtrainer/profiler/advanced.py +4 -4
  86. nshtrainer/profiler/pytorch.py +4 -4
  87. nshtrainer/profiler/simple.py +4 -4
  88. nshtrainer/trainer/__init__.py +1 -0
  89. nshtrainer/trainer/_config.py +164 -17
  90. nshtrainer/trainer/checkpoint_connector.py +23 -8
  91. nshtrainer/trainer/trainer.py +194 -76
  92. nshtrainer/util/_environment_info.py +21 -13
  93. nshtrainer/util/config/dtype.py +4 -4
  94. nshtrainer/util/typing_utils.py +1 -1
  95. {nshtrainer-0.44.1.dist-info → nshtrainer-1.0.0b10.dist-info}/METADATA +2 -2
  96. nshtrainer-1.0.0b10.dist-info/RECORD +143 -0
  97. nshtrainer/callbacks/_throughput_monitor_callback.py +0 -551
  98. nshtrainer/callbacks/throughput_monitor.py +0 -58
  99. nshtrainer/config/model/__init__.py +0 -41
  100. nshtrainer/config/model/base/__init__.py +0 -25
  101. nshtrainer/config/model/config/__init__.py +0 -37
  102. nshtrainer/config/model/mixins/logger/__init__.py +0 -22
  103. nshtrainer/config/runner/__init__.py +0 -22
  104. nshtrainer/ll/__init__.py +0 -59
  105. nshtrainer/ll/_experimental.py +0 -3
  106. nshtrainer/ll/actsave.py +0 -6
  107. nshtrainer/ll/callbacks.py +0 -3
  108. nshtrainer/ll/config.py +0 -6
  109. nshtrainer/ll/data.py +0 -3
  110. nshtrainer/ll/log.py +0 -5
  111. nshtrainer/ll/lr_scheduler.py +0 -3
  112. nshtrainer/ll/model.py +0 -21
  113. nshtrainer/ll/nn.py +0 -3
  114. nshtrainer/ll/optimizer.py +0 -3
  115. nshtrainer/ll/runner.py +0 -5
  116. nshtrainer/ll/snapshot.py +0 -3
  117. nshtrainer/ll/snoop.py +0 -3
  118. nshtrainer/ll/trainer.py +0 -3
  119. nshtrainer/ll/typecheck.py +0 -3
  120. nshtrainer/ll/util.py +0 -3
  121. nshtrainer/model/config.py +0 -218
  122. nshtrainer/runner.py +0 -101
  123. nshtrainer-0.44.1.dist-info/RECORD +0 -162
  124. {nshtrainer-0.44.1.dist-info → nshtrainer-1.0.0b10.dist-info}/WHEEL +0 -0
nshtrainer-0.44.1.dist-info/RECORD DELETED
@@ -1,162 +0,0 @@
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=EhA2f_WiZ79RcXL2nJbwCwNK620c8ugEVUmJ8CcVpsE,3723
120
- nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=gvUuv031lvWdXboDeH7iAF3ZgNPQK40bQwfmqb11TNk,5492
121
- nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=kOAlntpMKSO1fK94aocUcbR6wryW9tA9d74W_O7-KBM,2803
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.44.1.dist-info/METADATA,sha256=zIWwQb1PBF1WRQ5tlBmNa9uV0_BKwYFegZIfN-XSrx0,916
161
- nshtrainer-0.44.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
162
- nshtrainer-0.44.1.dist-info/RECORD,,