nshtrainer 0.44.0__py3-none-any.whl → 1.0.0b9__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 (125) 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 +51 -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/_base.py +2 -1
  79. nshtrainer/lr_scheduler/reduce_lr_on_plateau.py +9 -7
  80. nshtrainer/model/__init__.py +0 -4
  81. nshtrainer/model/base.py +64 -347
  82. nshtrainer/model/mixins/callback.py +24 -5
  83. nshtrainer/model/mixins/debug.py +86 -0
  84. nshtrainer/model/mixins/logger.py +142 -145
  85. nshtrainer/profiler/_base.py +2 -2
  86. nshtrainer/profiler/advanced.py +4 -4
  87. nshtrainer/profiler/pytorch.py +4 -4
  88. nshtrainer/profiler/simple.py +4 -4
  89. nshtrainer/trainer/__init__.py +1 -0
  90. nshtrainer/trainer/_config.py +164 -17
  91. nshtrainer/trainer/checkpoint_connector.py +23 -8
  92. nshtrainer/trainer/trainer.py +194 -76
  93. nshtrainer/util/_environment_info.py +21 -13
  94. nshtrainer/util/config/dtype.py +4 -4
  95. nshtrainer/util/typing_utils.py +1 -1
  96. {nshtrainer-0.44.0.dist-info → nshtrainer-1.0.0b9.dist-info}/METADATA +2 -2
  97. nshtrainer-1.0.0b9.dist-info/RECORD +143 -0
  98. nshtrainer/callbacks/_throughput_monitor_callback.py +0 -551
  99. nshtrainer/callbacks/throughput_monitor.py +0 -58
  100. nshtrainer/config/model/__init__.py +0 -41
  101. nshtrainer/config/model/base/__init__.py +0 -25
  102. nshtrainer/config/model/config/__init__.py +0 -37
  103. nshtrainer/config/model/mixins/logger/__init__.py +0 -22
  104. nshtrainer/config/runner/__init__.py +0 -22
  105. nshtrainer/ll/__init__.py +0 -59
  106. nshtrainer/ll/_experimental.py +0 -3
  107. nshtrainer/ll/actsave.py +0 -6
  108. nshtrainer/ll/callbacks.py +0 -3
  109. nshtrainer/ll/config.py +0 -6
  110. nshtrainer/ll/data.py +0 -3
  111. nshtrainer/ll/log.py +0 -5
  112. nshtrainer/ll/lr_scheduler.py +0 -3
  113. nshtrainer/ll/model.py +0 -21
  114. nshtrainer/ll/nn.py +0 -3
  115. nshtrainer/ll/optimizer.py +0 -3
  116. nshtrainer/ll/runner.py +0 -5
  117. nshtrainer/ll/snapshot.py +0 -3
  118. nshtrainer/ll/snoop.py +0 -3
  119. nshtrainer/ll/trainer.py +0 -3
  120. nshtrainer/ll/typecheck.py +0 -3
  121. nshtrainer/ll/util.py +0 -3
  122. nshtrainer/model/config.py +0 -218
  123. nshtrainer/runner.py +0 -101
  124. nshtrainer-0.44.0.dist-info/RECORD +0 -162
  125. {nshtrainer-0.44.0.dist-info → nshtrainer-1.0.0b9.dist-info}/WHEEL +0 -0
nshtrainer-1.0.0b9.dist-info/RECORD ADDED
@@ -0,0 +1,143 @@
1
+ nshtrainer/__init__.py,sha256=DR7rb_32mSztFqK8QFxtiEZPlE6XMWm1eCFv6uFl6sM,690
2
+ nshtrainer/_callback.py,sha256=tXQCDzS6CvMTuTY5lQSH5qZs1pXUi-gt9bQdpXMVdEs,12715
3
+ nshtrainer/_checkpoint/loader.py,sha256=dTYA5KTOlB7TqqD7UJzBLz2rEwYqpLMdBXwL6Qzt-n0,14112
4
+ nshtrainer/_checkpoint/metadata.py,sha256=sy3ualbRDHkEQmNm9nq7kUFG38QDqv3dbzhEPmH669U,5012
5
+ nshtrainer/_checkpoint/saver.py,sha256=LOP8jjKF0Dw9x9H-BKrLMWlEp1XTan2DUK0zQUCWw5U,1360
6
+ nshtrainer/_directory.py,sha256=p2uk1FnISFEpMqlDevKhoWhQsCEtvHUPg459K-86QA8,3053
7
+ nshtrainer/_experimental/__init__.py,sha256=U4S_2y3zgLZVfMenHRaJFBW8yqh2mUBuI291LGQVOJ8,35
8
+ nshtrainer/_hf_hub.py,sha256=xj91CEKcuvpaiipZbVZovZiU_fdsTkPXOkQ-3Xb-FhU,14183
9
+ nshtrainer/callbacks/__init__.py,sha256=eHN0sxFz1i3rYnSHdizK7cLTe5qYA8HQmN8xYc45FKk,3718
10
+ nshtrainer/callbacks/actsave.py,sha256=framZaRes5G-QjWgeDAJ72k3T3tFkj6Z7qmnlIUzLBQ,3759
11
+ nshtrainer/callbacks/base.py,sha256=JZ-WyIKeUYe8g2pLzs648WkxNfkTke7237n1LU_Ujto,3571
12
+ nshtrainer/callbacks/checkpoint/__init__.py,sha256=l8tkHc83_mLiU0-wT09SWdRzwpm2ulbkLzcuCmuTwzE,620
13
+ nshtrainer/callbacks/checkpoint/_base.py,sha256=ZVEUVl5kjCSSe69Q0rMUbKBNNUog0pxBwWkeyuxG2w0,6304
14
+ nshtrainer/callbacks/checkpoint/best_checkpoint.py,sha256=eyx2p48rFJK-gvgxzQx63suP-UmjJIrTeiTBhjG8A8w,2611
15
+ nshtrainer/callbacks/checkpoint/last_checkpoint.py,sha256=w-qys4xvx985gnoHwsWClpQBfqBe5XwojZo-UzHffcY,1157
16
+ nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py,sha256=G4amPXubdOD8DeAkrXz4O-rc73idd0niiJhyv-NjoJI,3458
17
+ nshtrainer/callbacks/debug_flag.py,sha256=85IG9bFn2-bUUiEAgvfeteUhYTEuSyPRv9BYgRqe8mc,1687
18
+ nshtrainer/callbacks/directory_setup.py,sha256=4Er9nOyNCMY9X27RzeW5z0Ky_OgnOBvxCF6xVsonMfg,2504
19
+ nshtrainer/callbacks/early_stopping.py,sha256=8Ai5aZLQrM9zMRUSFxfJ2Jdb8OoBE8ZWAcGSyw8nM-k,4715
20
+ nshtrainer/callbacks/ema.py,sha256=RbD6Mage7E2ke5-X4WcoOxMvNl4Z4_li2CnwmLFCCJc,12194
21
+ nshtrainer/callbacks/finite_checks.py,sha256=iCiKQ5i9RckkzcPeCHzC3hkg3AlW3ESuWtFwoF2RG0Q,2124
22
+ nshtrainer/callbacks/gradient_skipping.py,sha256=k5qNaNeileZ_5YFad4ssfLplMxMKeKFhPcY8-QVmLek,3464
23
+ nshtrainer/callbacks/interval.py,sha256=UCzUzt3XCFVyQyCWL9lOrStkkxesvduNOYk8yMrGTTk,8116
24
+ nshtrainer/callbacks/log_epoch.py,sha256=Wr-Ksxsynsqu_zyB_zoiPLjnWv-ksC3xPekY6iyN-P8,1396
25
+ nshtrainer/callbacks/norm_logging.py,sha256=C44Mvt73gqQEpCFd0j3qYg6NY7sL2jm3X1qJVY_XLfI,6329
26
+ nshtrainer/callbacks/print_table.py,sha256=WIgfzVSfAfS3_8kUuX-nWJOGWBEmtNlejypuoJQViPY,2884
27
+ nshtrainer/callbacks/rlp_sanity_checks.py,sha256=kWl2dYOXn2L8k6ub_012jNkqOxtyea1yr1qWRNG6UW4,9990
28
+ nshtrainer/callbacks/shared_parameters.py,sha256=33eRzifNj6reKbvmGuam1hUofo3sD4J6Zz8gtoHK93I,2977
29
+ nshtrainer/callbacks/timer.py,sha256=BB-M7tV4QNYOwY_Su6j9P7IILxVRae_upmDq4qsxiao,4670
30
+ nshtrainer/callbacks/wandb_upload_code.py,sha256=PTqNE1QB5U8NR5zhbiQZrmQuugX2UV7B12UdMpo9aV0,2353
31
+ nshtrainer/callbacks/wandb_watch.py,sha256=tTTcFzxd2Ia9xu8tCogQ5CLJZBq1ne5JlpGVE75vKYs,2976
32
+ nshtrainer/config/__init__.py,sha256=0TczWa5OFRKOGKHgabeB7VUMxPpD0RCgDR6AvdAD-tI,22721
33
+ nshtrainer/config/_checkpoint/__init__.py,sha256=vuiBbd4VzCo7lRyhyTUArEQeWwJkewvNPKDxBJiUHoY,2719
34
+ nshtrainer/config/_checkpoint/loader/__init__.py,sha256=hdLpypoEkES1MTaTHAdGFJnSoZzgx_8NzAKbK143SyI,2399
35
+ nshtrainer/config/_checkpoint/metadata/__init__.py,sha256=X9KxpcoHQbJp6-MTGvp4pct-MYHaHcl82s9yqZ5KiSk,867
36
+ nshtrainer/config/_directory/__init__.py,sha256=mTUoSz-DSsvI2M98cqu2Z2x215oM0sLyljh_5rVexvQ,1029
37
+ nshtrainer/config/_hf_hub/__init__.py,sha256=3HGCGhRb7NhOuLeskGqbYNuS9c81oOUbX6ibyF3XiCY,1063
38
+ nshtrainer/config/callbacks/__init__.py,sha256=-dHN8NZdCaNUy_isnlh779FZ1w9_WkOkv6VSN_-86jM,7316
39
+ nshtrainer/config/callbacks/actsave/__init__.py,sha256=AkVWS9vCcDJFpPUpyc7i9cjaFZU2kKxDyFDqakMZA-E,809
40
+ nshtrainer/config/callbacks/base/__init__.py,sha256=OdtHDMkYC_ioCEAkg7bSQi3o7e2t5WHPcFjavXdfdTA,602
41
+ nshtrainer/config/callbacks/checkpoint/__init__.py,sha256=DE-JTtY4NdaP3mgWduearFYMvy1tswWRBWMde06RzQc,2700
42
+ nshtrainer/config/callbacks/checkpoint/_base/__init__.py,sha256=PNCnWgzHzg5yHDLSoTUgcuyZbBDc8wHfug8hS7xdfFg,1255
43
+ nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py,sha256=ZEBXeXL7IFztb8O8GRjSe8R62jxxNxVT5pQ2AJdMvQA,1637
44
+ nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py,sha256=p3BY0jurlt5mnenZ07Pi0dBILtVQrucXw8lfIx0upUQ,1355
45
+ nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py,sha256=WQ09bjj1E_iq_XDILa9Kwe9WrUxrUjz5sKy6IFb14p4,1069
46
+ nshtrainer/config/callbacks/debug_flag/__init__.py,sha256=z0hzh1q83XJ1o6p26QTfyCjpMcHERn--ovts1MzQVGI,908
47
+ nshtrainer/config/callbacks/directory_setup/__init__.py,sha256=L2fK_nQkEky9TrqFmZwW1XEqsftn_i8xw_4uDcDXp0k,965
48
+ nshtrainer/config/callbacks/early_stopping/__init__.py,sha256=UVj9fsaS6dDmKM2qIVJwtE8HrcxmTper-eU6FyAeFps,1198
49
+ nshtrainer/config/callbacks/ema/__init__.py,sha256=KlPGdJWjYTKLdpl-VnN4BYY2sA_LX3rsoZo-4Q9pbv4,809
50
+ nshtrainer/config/callbacks/finite_checks/__init__.py,sha256=V6Owp05XdIk3EO67AMVGdwbT4-D86QRuvqWM2gu5Xpw,949
51
+ nshtrainer/config/callbacks/gradient_skipping/__init__.py,sha256=RhIfJFq-x_sWYrWVGaVEBeT8uUFYjFgt0Ug8pPgpJSg,981
52
+ nshtrainer/config/callbacks/log_epoch/__init__.py,sha256=4jePzjE3bVxaI7hQrcWW5SrKT5MrFyplJZwK8bQHbGI,900
53
+ nshtrainer/config/callbacks/norm_logging/__init__.py,sha256=8M9hcGpEuQadfgTR4-YL4TWeyxZjg0s84x420B03-aE,941
54
+ nshtrainer/config/callbacks/print_table/__init__.py,sha256=Ni47iS2mIzwGu8XuHfUY5BJKawUO_2TyJMZ62QBpEW0,961
55
+ nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py,sha256=OgPywk8Z9y_dnq_liH2PPWuQSpUlQ_Q2-q99HDN9Leg,977
56
+ nshtrainer/config/callbacks/shared_parameters/__init__.py,sha256=1wvcw4hu7q2CDT6zJi6nb2Ixnz0QSwKJ0a6w5dKxJ5M,981
57
+ nshtrainer/config/callbacks/timer/__init__.py,sha256=LE8JwBK9wxfdloYHfwMfg4yIGv4V-zd6gtFde6ngubs,892
58
+ nshtrainer/config/callbacks/wandb_upload_code/__init__.py,sha256=czWDuQtQR9eWyQEolEUvGxjMuCKkFlb1RSVrPP_bIdU,977
59
+ nshtrainer/config/callbacks/wandb_watch/__init__.py,sha256=y3t2NUFkXpfwfb3aKXqOWyzaWHhhaphMY-TkVyiipD0,933
60
+ nshtrainer/config/loggers/__init__.py,sha256=mESJtuNsxdS3HmijNSFPwoW3k6CDrlcXi4BYcCZopno,2528
61
+ nshtrainer/config/loggers/_base/__init__.py,sha256=CFo8APn-VRiUI19lcSV97icCJrKefQ1wAFuZTWFn_fQ,562
62
+ nshtrainer/config/loggers/actsave/__init__.py,sha256=TLDj9Lexs6P2K-yB43MQ0z2XN9jOCrOc34c0DEJ1I_4,847
63
+ nshtrainer/config/loggers/csv/__init__.py,sha256=yHEKSpdxU67jm-TVBdFlsJ6Gvugxl3QZkT5lbuKe3ek,755
64
+ nshtrainer/config/loggers/tensorboard/__init__.py,sha256=sw2lvaAJLxFQze5fxY-WJfJHHGZ2RX3VpZ-oqW9a_Ck,896
65
+ nshtrainer/config/loggers/wandb/__init__.py,sha256=siElybkmRptnhJkAZTAqzeYNrTV8Yj2YeGBfYMM7Vpc,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=6sqqrymsn9c_UEShRGvgLTie8maP8lX3UH5Ee2l7jZM,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/nn/__init__.py,sha256=XV_hbkQyEafbctOnPl-XU97KPQJTfQVLRSchEnzfRn4,3990
73
+ nshtrainer/config/nn/mlp/__init__.py,sha256=C6dft6w162_J3YggaC5JmpnEjPUcbiw4lSu9FFB0SPU,934
74
+ nshtrainer/config/nn/nonlinearity/__init__.py,sha256=_njAmDmI3AI7lMJgUOFaD7KmmEHWQJICASWeiQkPghI,4797
75
+ nshtrainer/config/optimizer/__init__.py,sha256=UQ6ePMmJ9sdj9UA4LQRe84k95GtJhVl7lgKQfXCCEbc,936
76
+ nshtrainer/config/profiler/__init__.py,sha256=61p5X6lNjW3oPO6ay05eunYpHz-shY5yHCIG0fhaUD0,1521
77
+ nshtrainer/config/profiler/_base/__init__.py,sha256=smDYCplrI5B38XJcNZ462ZeTo9l1ieLXCxRxf-FJEUY,602
78
+ nshtrainer/config/profiler/advanced/__init__.py,sha256=VCCvbzhEeOcdZ0Unvk_anAcmbQuGogTQhK_bXs5RG9U,892
79
+ nshtrainer/config/profiler/pytorch/__init__.py,sha256=hJ90ym5ElI-BY_XS3VSLjcgQWfV0Pp1MdzTU6Qi8MFg,884
80
+ nshtrainer/config/profiler/simple/__init__.py,sha256=18V64kKYrJeSCrPmY3wYnshEISaf7xmrfw2Ny-6P3uE,859
81
+ nshtrainer/config/trainer/__init__.py,sha256=Gf5RizrVL84NWfHnagVCRHtXiD6x0UA4N7vhYypluTk,7916
82
+ nshtrainer/config/trainer/_config/__init__.py,sha256=BeQ5t_9d6rx6SbSu4ZqD9eitLCQRpTOVOnSxT0LCrlM,7806
83
+ nshtrainer/config/trainer/checkpoint_connector/__init__.py,sha256=pSu79zOFFWvqjI3SkHWl13H8ZNJFTc6a5J1r2KnfUKM,667
84
+ nshtrainer/config/trainer/trainer/__init__.py,sha256=P-Y2DOZZcJtEdPjGEKCxq5R3JSzKhUUoidkSvO_cfKI,797
85
+ nshtrainer/config/util/__init__.py,sha256=ZcmEqg2OWKKcPBqzDG1SnuaAMgR4_c0jog-Xg6QTUzc,4555
86
+ nshtrainer/config/util/_environment_info/__init__.py,sha256=MQJnvrGBRotjN0wuo7ntyfHtu45mxZuo2i5kEtgqheI,3730
87
+ nshtrainer/config/util/config/__init__.py,sha256=1fssbW-Z5o0Ckn1L3g22bhT2kFg6OLwyqETq5gCJo4w,1162
88
+ nshtrainer/config/util/config/dtype/__init__.py,sha256=NCXMVO-EUz3JvPmlDci72O9ZeqpCANs0ieeoXAu9Ljw,550
89
+ nshtrainer/config/util/config/duration/__init__.py,sha256=8llT1MCKQpsdNldN5h5Wo0GjUuRn28Sxw2FTXTNKBpM,1060
90
+ nshtrainer/data/__init__.py,sha256=K4i3Tw4g9EOK2zlMMbidi99y0SyI4w8P7_XUf1n42Ts,260
91
+ nshtrainer/data/balanced_batch_sampler.py,sha256=r1cBKRXKHD8E1Ax6tj-FUbE-z1qpbO58mQ9VrK9uLnc,5481
92
+ nshtrainer/data/datamodule.py,sha256=fVPAZQekPGHCdtFLHgyeBuB4lF35LlEq3f9u02EBvG4,1810
93
+ nshtrainer/data/transform.py,sha256=qd0lIocO59Fk_m90xyOHgFezbymd1mRwly8nbYIfHGc,2263
94
+ nshtrainer/loggers/__init__.py,sha256=11X6D_lF0cHAkxkYsVZY3Q3r30Fq0FUi9heeb5RD870,570
95
+ nshtrainer/loggers/_base.py,sha256=nw4AZzJP3Z-fljgQlgq7FkuMkPmYKTsXj7OfJJSmtXI,811
96
+ nshtrainer/loggers/actsave.py,sha256=23Kre-mq-Y9Iw1SRyGmHnHK1bc_0gTWFXJViv9bkVz0,1324
97
+ nshtrainer/loggers/csv.py,sha256=Deh5gm3oROJbQzigV4SHni5JRwSrBdm-4YD3yrcGnHo,1104
98
+ nshtrainer/loggers/tensorboard.py,sha256=jP9V4nlq_MXUaoD6xv1Cws2ioft83Lm8yUJhhGhuUrQ,2268
99
+ nshtrainer/loggers/wandb.py,sha256=EjKQQznLSUSCWO7uIviz9g0dVW4ZLxb_8UVhY4vR7r0,6800
100
+ nshtrainer/lr_scheduler/__init__.py,sha256=o7eDaj-KOetQvEwMjGKpcQofs1jimqUzHbJ5YM5cOMY,772
101
+ nshtrainer/lr_scheduler/_base.py,sha256=EhA2f_WiZ79RcXL2nJbwCwNK620c8ugEVUmJ8CcVpsE,3723
102
+ nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=gvUuv031lvWdXboDeH7iAF3ZgNPQK40bQwfmqb11TNk,5492
103
+ nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=vXH5S26ESHO_LPPqW8aDC3S5NGoZYkXeFjAOgttaUX8,2870
104
+ nshtrainer/metrics/__init__.py,sha256=Nqkn_jsDf3n5WtfMcnaaEftYjIIT2b-S7rmsB1MOMkU,86
105
+ nshtrainer/metrics/_config.py,sha256=kR9OJLZXIaZpG8UpHG3EQwOL36HB8yPk6afYjoIM0XM,1324
106
+ nshtrainer/model/__init__.py,sha256=3G-bwPPSRStWdsdwG9-rn0bXcRpEiP1BiQpF_qavtls,97
107
+ nshtrainer/model/base.py,sha256=qJXzu1w4mqF1eVwRBahEuJwJTIVSAZGZaV1vigwrg0Y,6817
108
+ nshtrainer/model/mixins/callback.py,sha256=Ea_legORzs0N078j0N9RJivDVeWH5KtXDpdJS75IwIo,3098
109
+ nshtrainer/model/mixins/debug.py,sha256=1LX9KzeFX9JDPs_a6YCdYDZXLhEk_5rBO2aCqlfBy7w,2087
110
+ nshtrainer/model/mixins/logger.py,sha256=LQDJJbiv30PlWX6rTT_EhjNBNfUFfcvGz5sX4MnOCzI,5330
111
+ nshtrainer/nn/__init__.py,sha256=sANhrZpeN5syLKOsmXMwhaFl2SBFPWcLaEe1EH22TWQ,1463
112
+ nshtrainer/nn/mlp.py,sha256=2W8bzE96DzCMzGm6WPiPhNFQfhqaoG3GXPn_oKBnlUM,5988
113
+ nshtrainer/nn/module_dict.py,sha256=9plb8aQUx5TUEPhX5jI9u8LrpTeKe7jZAHi8iIqcN8w,2365
114
+ nshtrainer/nn/module_list.py,sha256=UB43pcwD_3nUke_DyLQt-iXKhWdKM6Zjm84lRC1hPYA,1755
115
+ nshtrainer/nn/nonlinearity.py,sha256=mp5XvXRHURB6jwuZ0YyTj5ZoHJYNJNgO2aLtUY1D-2Y,6114
116
+ nshtrainer/optimizer.py,sha256=73UWfuTU4moDQNSKI8VSQSDc6_i7mBtemxvhBkM-fC4,1563
117
+ nshtrainer/profiler/__init__.py,sha256=X6E4cbIRaXHCDgQtT3LFPw5xIRBhqnJ03SVFbsD-Yd8,516
118
+ nshtrainer/profiler/_base.py,sha256=kFcSVn9gJuMwgDxbfyHh46CmEAIPZjxw3yjPbKgzvwA,950
119
+ nshtrainer/profiler/advanced.py,sha256=XrM3FX0ThCv5UwUrrH0l4Ow4LGAtpiBww2N8QAU5NOQ,1160
120
+ nshtrainer/profiler/pytorch.py,sha256=8K37XvPnCApUpIK8tA2zNMFIaIiTLSoxKQoiyCPBm1Q,2757
121
+ nshtrainer/profiler/simple.py,sha256=PimjqcU-JuS-8C0ZGHAdwCxgNLij4x0FH6WXsjBQzZs,1005
122
+ nshtrainer/scripts/find_packages.py,sha256=-oNnSNPp3pujCVgManW_LFlJcnvhrtHgvUJ4W88-6-o,1460
123
+ nshtrainer/trainer/__init__.py,sha256=MmoydVS6aYeav7zgDAUHxAQrV_PMQsbnZTCuPnLH9Wk,128
124
+ nshtrainer/trainer/_config.py,sha256=5HC2Gs7YbJK9gMcKm4cQ78B3PxcOvwQ372Glhf4fzbo,34232
125
+ nshtrainer/trainer/_runtime_callback.py,sha256=T3epaj1YeIN0R8CS2cg5HNJIB21TyaD_PVNNOPJ6nJs,4200
126
+ nshtrainer/trainer/checkpoint_connector.py,sha256=pC1tTDcq0p6sAsoTmAbwINW49IfqupMMtnE9-AKdTUw,2824
127
+ nshtrainer/trainer/signal_connector.py,sha256=YMJf6vTnW0JcnBkuYikm9x_9XscaokrCEzCn4THOGao,10776
128
+ nshtrainer/trainer/trainer.py,sha256=JLcfdPqlOaAmKJAE3w9xIEai7Xyyl3JhDyMPsZ8fyv8,20692
129
+ nshtrainer/util/_environment_info.py,sha256=MT8mBe6ZolRfKiwU-les1P-lPNPqXpHQcfADrh_A3uY,24629
130
+ nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
131
+ nshtrainer/util/bf16.py,sha256=9QhHZCkYSfYpIcxwAMoXyuh2yTSHBzT-EdLQB297jEs,762
132
+ nshtrainer/util/config/__init__.py,sha256=Z39JJufSb61Lhn2GfVcv3eFW_eorOrN9-9llDWlnZZM,272
133
+ nshtrainer/util/config/dtype.py,sha256=VQ02dbouSehCdziFN7NfsEu7o_dcW5SNbl5bhNIwn30,2021
134
+ nshtrainer/util/config/duration.py,sha256=mM-UfU_HvhXwW33TYEDg0x58n80tnle2e6VaWtxZTjk,764
135
+ nshtrainer/util/environment.py,sha256=s-B5nY0cKYXdFMdNYumvC_xxacMATiI4DvV2gUDu20k,4195
136
+ nshtrainer/util/path.py,sha256=mfuMRk29YpmwdRsyyf99VT0n_T9caM99SS7VfA6oIHE,3501
137
+ nshtrainer/util/seed.py,sha256=diMV8iwBKN7Xxt5pELmui-gyqyT80_CZzomrWhNss0k,316
138
+ nshtrainer/util/slurm.py,sha256=HflkP5iI_r4UHMyPjw9R4dD5AHsJUpcfJw5PLvGYBRM,1603
139
+ nshtrainer/util/typed.py,sha256=Xt5fUU6zwLKSTLUdenovnKK0N8qUq89Kddz2_XeykVQ,164
140
+ nshtrainer/util/typing_utils.py,sha256=MjY-CUX9R5Tzat-BlFnQjwl1PQ_W2yZQoXhkYHlJ_VA,442
141
+ nshtrainer-1.0.0b9.dist-info/METADATA,sha256=X_GVytB3jZ2QlxKmMtttbd9S0tz9TMyWSmaJHAy11Q4,936
142
+ nshtrainer-1.0.0b9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
143
+ nshtrainer-1.0.0b9.dist-info/RECORD,,