nshtrainer 0.33.0__tar.gz → 0.33.2__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 (100) hide show
  1. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/PKG-INFO +1 -1
  2. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/pyproject.toml +1 -1
  3. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/config.py +2 -0
  4. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/README.md +0 -0
  5. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/__init__.py +0 -0
  6. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/_callback.py +0 -0
  7. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/_checkpoint/loader.py +0 -0
  8. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/_checkpoint/metadata.py +0 -0
  9. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/_checkpoint/saver.py +0 -0
  10. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/_directory.py +0 -0
  11. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/_experimental/__init__.py +0 -0
  12. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/_hf_hub.py +0 -0
  13. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/__init__.py +0 -0
  14. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/_throughput_monitor_callback.py +0 -0
  15. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/actsave.py +0 -0
  16. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/base.py +0 -0
  17. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/checkpoint/__init__.py +0 -0
  18. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/checkpoint/_base.py +0 -0
  19. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/checkpoint/best_checkpoint.py +0 -0
  20. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/checkpoint/last_checkpoint.py +0 -0
  21. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py +0 -0
  22. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/debug_flag.py +0 -0
  23. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/directory_setup.py +0 -0
  24. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/early_stopping.py +0 -0
  25. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/ema.py +0 -0
  26. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/finite_checks.py +0 -0
  27. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/gradient_skipping.py +0 -0
  28. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/interval.py +0 -0
  29. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/log_epoch.py +0 -0
  30. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/norm_logging.py +0 -0
  31. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/print_table.py +0 -0
  32. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/rlp_sanity_checks.py +0 -0
  33. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/shared_parameters.py +0 -0
  34. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/throughput_monitor.py +0 -0
  35. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/timer.py +0 -0
  36. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/callbacks/wandb_watch.py +0 -0
  37. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/data/__init__.py +0 -0
  38. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/data/balanced_batch_sampler.py +0 -0
  39. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/data/transform.py +0 -0
  40. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/__init__.py +0 -0
  41. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/_experimental.py +0 -0
  42. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/actsave.py +0 -0
  43. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/callbacks.py +0 -0
  44. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/config.py +0 -0
  45. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/data.py +0 -0
  46. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/log.py +0 -0
  47. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/lr_scheduler.py +0 -0
  48. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/model.py +0 -0
  49. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/nn.py +0 -0
  50. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/optimizer.py +0 -0
  51. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/runner.py +0 -0
  52. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/snapshot.py +0 -0
  53. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/snoop.py +0 -0
  54. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/trainer.py +0 -0
  55. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/typecheck.py +0 -0
  56. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/ll/util.py +0 -0
  57. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/loggers/__init__.py +0 -0
  58. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/loggers/_base.py +0 -0
  59. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/loggers/csv.py +0 -0
  60. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/loggers/tensorboard.py +0 -0
  61. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/loggers/wandb.py +0 -0
  62. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/lr_scheduler/__init__.py +0 -0
  63. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/lr_scheduler/_base.py +0 -0
  64. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/lr_scheduler/linear_warmup_cosine.py +0 -0
  65. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/lr_scheduler/reduce_lr_on_plateau.py +0 -0
  66. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/metrics/__init__.py +0 -0
  67. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/metrics/_config.py +0 -0
  68. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/model/__init__.py +0 -0
  69. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/model/base.py +0 -0
  70. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/model/config.py +0 -0
  71. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/model/mixins/logger.py +0 -0
  72. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/nn/__init__.py +0 -0
  73. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/nn/mlp.py +0 -0
  74. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/nn/module_dict.py +0 -0
  75. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/nn/module_list.py +0 -0
  76. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/nn/nonlinearity.py +0 -0
  77. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/optimizer.py +0 -0
  78. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/profiler/__init__.py +0 -0
  79. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/profiler/_base.py +0 -0
  80. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/profiler/advanced.py +0 -0
  81. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/profiler/pytorch.py +0 -0
  82. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/profiler/simple.py +0 -0
  83. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/runner.py +0 -0
  84. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/scripts/find_packages.py +0 -0
  85. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/trainer/__init__.py +0 -0
  86. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/trainer/_config.py +0 -0
  87. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/trainer/_runtime_callback.py +0 -0
  88. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/trainer/checkpoint_connector.py +0 -0
  89. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/trainer/signal_connector.py +0 -0
  90. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/trainer/trainer.py +0 -0
  91. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/_environment_info.py +0 -0
  92. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/_useful_types.py +0 -0
  93. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/config/__init__.py +0 -0
  94. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/config/duration.py +0 -0
  95. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/environment.py +0 -0
  96. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/path.py +0 -0
  97. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/seed.py +0 -0
  98. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/slurm.py +0 -0
  99. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/typed.py +0 -0
  100. {nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/util/typing_utils.py +0 -0
{nshtrainer-0.33.0 → nshtrainer-0.33.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.33.0
3
+ Version: 0.33.2
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.33.0 → nshtrainer-0.33.2}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "nshtrainer"
3
- version = "0.33.0"
3
+ version = "0.33.2"
4
4
  description = ""
5
5
  authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
6
6
  readme = "README.md"
{nshtrainer-0.33.0 → nshtrainer-0.33.2}/src/nshtrainer/config.py RENAMED
@@ -59,6 +59,7 @@ from nshtrainer.loggers.tensorboard import (
59
59
  TensorboardLoggerConfig as TensorboardLoggerConfig,
60
60
  )
61
61
  from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
62
+ from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
62
63
  from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase
63
64
  from nshtrainer.lr_scheduler.linear_warmup_cosine import (
64
65
  DurationConfig as DurationConfig,
@@ -103,6 +104,7 @@ from nshtrainer.nn.nonlinearity import (
103
104
  )
104
105
  from nshtrainer.nn.nonlinearity import TanhNonlinearityConfig as TanhNonlinearityConfig
105
106
  from nshtrainer.optimizer import AdamWConfig as AdamWConfig
107
+ from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
106
108
  from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
107
109
  from nshtrainer.profiler._base import BaseProfilerConfig as BaseProfilerConfig
108
110
  from nshtrainer.profiler.advanced import (
{nshtrainer-0.33.0 → nshtrainer-0.33.2}/README.md RENAMED
File without changes