nshtrainer 0.37.0__tar.gz → 0.38.0__tar.gz

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 (104) hide show
  1. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/PKG-INFO +1 -1
  2. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/pyproject.toml +1 -1
  3. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/config.py +7 -7
  4. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/README.md +0 -0
  5. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/__init__.py +0 -0
  6. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/_callback.py +0 -0
  7. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/_checkpoint/loader.py +0 -0
  8. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/_checkpoint/metadata.py +0 -0
  9. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/_checkpoint/saver.py +0 -0
  10. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/_directory.py +0 -0
  11. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/_experimental/__init__.py +0 -0
  12. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/_hf_hub.py +0 -0
  13. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/__init__.py +0 -0
  14. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/_throughput_monitor_callback.py +0 -0
  15. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/actsave.py +0 -0
  16. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/base.py +0 -0
  17. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/checkpoint/__init__.py +0 -0
  18. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/checkpoint/_base.py +0 -0
  19. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/checkpoint/best_checkpoint.py +0 -0
  20. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/checkpoint/last_checkpoint.py +0 -0
  21. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py +0 -0
  22. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/debug_flag.py +0 -0
  23. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/directory_setup.py +0 -0
  24. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/early_stopping.py +0 -0
  25. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/ema.py +0 -0
  26. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/finite_checks.py +0 -0
  27. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/gradient_skipping.py +0 -0
  28. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/interval.py +0 -0
  29. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/log_epoch.py +0 -0
  30. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/norm_logging.py +0 -0
  31. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/print_table.py +0 -0
  32. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/rlp_sanity_checks.py +0 -0
  33. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/shared_parameters.py +0 -0
  34. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/throughput_monitor.py +0 -0
  35. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/timer.py +0 -0
  36. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/wandb_upload_code.py +0 -0
  37. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/callbacks/wandb_watch.py +0 -0
  38. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/data/__init__.py +0 -0
  39. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/data/balanced_batch_sampler.py +0 -0
  40. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/data/transform.py +0 -0
  41. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/__init__.py +0 -0
  42. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/_experimental.py +0 -0
  43. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/actsave.py +0 -0
  44. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/callbacks.py +0 -0
  45. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/config.py +0 -0
  46. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/data.py +0 -0
  47. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/log.py +0 -0
  48. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/lr_scheduler.py +0 -0
  49. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/model.py +0 -0
  50. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/nn.py +0 -0
  51. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/optimizer.py +0 -0
  52. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/runner.py +0 -0
  53. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/snapshot.py +0 -0
  54. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/snoop.py +0 -0
  55. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/trainer.py +0 -0
  56. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/typecheck.py +0 -0
  57. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/ll/util.py +0 -0
  58. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/loggers/__init__.py +0 -0
  59. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/loggers/_base.py +0 -0
  60. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/loggers/csv.py +0 -0
  61. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/loggers/tensorboard.py +0 -0
  62. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/loggers/wandb.py +0 -0
  63. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/lr_scheduler/__init__.py +0 -0
  64. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/lr_scheduler/_base.py +0 -0
  65. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/lr_scheduler/linear_warmup_cosine.py +0 -0
  66. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/lr_scheduler/reduce_lr_on_plateau.py +0 -0
  67. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/metrics/__init__.py +0 -0
  68. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/metrics/_config.py +0 -0
  69. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/model/__init__.py +0 -0
  70. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/model/base.py +0 -0
  71. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/model/config.py +0 -0
  72. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/model/mixins/callback.py +0 -0
  73. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/model/mixins/logger.py +0 -0
  74. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/nn/__init__.py +0 -0
  75. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/nn/mlp.py +0 -0
  76. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/nn/module_dict.py +0 -0
  77. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/nn/module_list.py +0 -0
  78. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/nn/nonlinearity.py +0 -0
  79. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/optimizer.py +0 -0
  80. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/profiler/__init__.py +0 -0
  81. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/profiler/_base.py +0 -0
  82. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/profiler/advanced.py +0 -0
  83. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/profiler/pytorch.py +0 -0
  84. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/profiler/simple.py +0 -0
  85. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/runner.py +0 -0
  86. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/scripts/find_packages.py +0 -0
  87. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/trainer/__init__.py +0 -0
  88. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/trainer/_config.py +0 -0
  89. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/trainer/_runtime_callback.py +0 -0
  90. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/trainer/checkpoint_connector.py +0 -0
  91. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/trainer/signal_connector.py +0 -0
  92. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/trainer/trainer.py +0 -0
  93. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/_environment_info.py +0 -0
  94. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/_useful_types.py +0 -0
  95. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/bf16.py +0 -0
  96. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/config/__init__.py +0 -0
  97. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/config/dtype.py +0 -0
  98. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/config/duration.py +0 -0
  99. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/environment.py +0 -0
  100. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/path.py +0 -0
  101. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/seed.py +0 -0
  102. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/slurm.py +0 -0
  103. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/typed.py +0 -0
  104. {nshtrainer-0.37.0 → nshtrainer-0.38.0}/src/nshtrainer/util/typing_utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.37.0
3
+ Version: 0.38.0
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "nshtrainer"
3
- version = "0.37.0"
3
+ version = "0.38.0"
4
4
  description = ""
5
5
  authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
6
6
  readme = "README.md"
@@ -65,13 +65,13 @@ from nshtrainer.callbacks.wandb_upload_code import (
65
65
  WandbUploadCodeConfig as WandbUploadCodeConfig,
66
66
  )
67
67
  from nshtrainer.callbacks.wandb_watch import WandbWatchConfig as WandbWatchConfig
68
- from nshtrainer.config import LRSchedulerConfig as LRSchedulerConfig
69
68
  from nshtrainer.loggers._base import BaseLoggerConfig as BaseLoggerConfig
70
69
  from nshtrainer.loggers.csv import CSVLoggerConfig as CSVLoggerConfig
71
70
  from nshtrainer.loggers.tensorboard import (
72
71
  TensorboardLoggerConfig as TensorboardLoggerConfig,
73
72
  )
74
73
  from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
74
+ from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
75
75
  from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase
76
76
  from nshtrainer.lr_scheduler.linear_warmup_cosine import (
77
77
  DurationConfig as DurationConfig,
@@ -95,10 +95,10 @@ from nshtrainer.nn.nonlinearity import MishNonlinearityConfig as MishNonlinearit
95
95
  from nshtrainer.nn.nonlinearity import NonlinearityConfig as NonlinearityConfig
96
96
  from nshtrainer.nn.nonlinearity import PReLUConfig as PReLUConfig
97
97
  from nshtrainer.nn.nonlinearity import ReLUNonlinearityConfig as ReLUNonlinearityConfig
98
+ from nshtrainer.nn.nonlinearity import SiLUNonlinearityConfig as SiLUNonlinearityConfig
98
99
  from nshtrainer.nn.nonlinearity import (
99
100
  SigmoidNonlinearityConfig as SigmoidNonlinearityConfig,
100
101
  )
101
- from nshtrainer.nn.nonlinearity import SiLUNonlinearityConfig as SiLUNonlinearityConfig
102
102
  from nshtrainer.nn.nonlinearity import (
103
103
  SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig,
104
104
  )
@@ -138,12 +138,12 @@ from nshtrainer.trainer._config import ReproducibilityConfig as ReproducibilityC
138
138
  from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
139
139
  from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
140
140
  from nshtrainer.util._environment_info import (
141
- EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
141
+ EnvironmentCUDAConfig as EnvironmentCUDAConfig,
142
142
  )
143
- from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
144
143
  from nshtrainer.util._environment_info import (
145
- EnvironmentCUDAConfig as EnvironmentCUDAConfig,
144
+ EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
146
145
  )
146
+ from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
147
147
  from nshtrainer.util._environment_info import (
148
148
  EnvironmentGPUConfig as EnvironmentGPUConfig,
149
149
  )
@@ -151,10 +151,10 @@ from nshtrainer.util._environment_info import (
151
151
  EnvironmentHardwareConfig as EnvironmentHardwareConfig,
152
152
  )
153
153
  from nshtrainer.util._environment_info import (
154
- EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
154
+ EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig,
155
155
  )
156
156
  from nshtrainer.util._environment_info import (
157
- EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig,
157
+ EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
158
158
  )
159
159
  from nshtrainer.util._environment_info import (
160
160
  EnvironmentPackageConfig as EnvironmentPackageConfig,
File without changes