nshtrainer 0.33.0__tar.gz → 0.33.1__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.1}/PKG-INFO +1 -1
  2. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/pyproject.toml +1 -1
  3. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/config.py +37 -0
  4. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/README.md +0 -0
  5. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/__init__.py +0 -0
  6. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/_callback.py +0 -0
  7. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/_checkpoint/loader.py +0 -0
  8. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/_checkpoint/metadata.py +0 -0
  9. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/_checkpoint/saver.py +0 -0
  10. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/_directory.py +0 -0
  11. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/_experimental/__init__.py +0 -0
  12. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/_hf_hub.py +0 -0
  13. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/__init__.py +0 -0
  14. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/_throughput_monitor_callback.py +0 -0
  15. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/actsave.py +0 -0
  16. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/base.py +0 -0
  17. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/checkpoint/__init__.py +0 -0
  18. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/checkpoint/_base.py +0 -0
  19. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/checkpoint/best_checkpoint.py +0 -0
  20. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/checkpoint/last_checkpoint.py +0 -0
  21. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/checkpoint/on_exception_checkpoint.py +0 -0
  22. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/debug_flag.py +0 -0
  23. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/directory_setup.py +0 -0
  24. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/early_stopping.py +0 -0
  25. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/ema.py +0 -0
  26. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/finite_checks.py +0 -0
  27. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/gradient_skipping.py +0 -0
  28. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/interval.py +0 -0
  29. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/log_epoch.py +0 -0
  30. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/norm_logging.py +0 -0
  31. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/print_table.py +0 -0
  32. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/rlp_sanity_checks.py +0 -0
  33. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/shared_parameters.py +0 -0
  34. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/throughput_monitor.py +0 -0
  35. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/timer.py +0 -0
  36. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/callbacks/wandb_watch.py +0 -0
  37. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/data/__init__.py +0 -0
  38. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/data/balanced_batch_sampler.py +0 -0
  39. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/data/transform.py +0 -0
  40. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/__init__.py +0 -0
  41. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/_experimental.py +0 -0
  42. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/actsave.py +0 -0
  43. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/callbacks.py +0 -0
  44. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/config.py +0 -0
  45. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/data.py +0 -0
  46. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/log.py +0 -0
  47. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/lr_scheduler.py +0 -0
  48. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/model.py +0 -0
  49. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/nn.py +0 -0
  50. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/optimizer.py +0 -0
  51. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/runner.py +0 -0
  52. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/snapshot.py +0 -0
  53. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/snoop.py +0 -0
  54. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/trainer.py +0 -0
  55. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/typecheck.py +0 -0
  56. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/ll/util.py +0 -0
  57. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/loggers/__init__.py +0 -0
  58. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/loggers/_base.py +0 -0
  59. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/loggers/csv.py +0 -0
  60. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/loggers/tensorboard.py +0 -0
  61. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/loggers/wandb.py +0 -0
  62. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/lr_scheduler/__init__.py +0 -0
  63. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/lr_scheduler/_base.py +0 -0
  64. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/lr_scheduler/linear_warmup_cosine.py +0 -0
  65. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/lr_scheduler/reduce_lr_on_plateau.py +0 -0
  66. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/metrics/__init__.py +0 -0
  67. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/metrics/_config.py +0 -0
  68. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/model/__init__.py +0 -0
  69. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/model/base.py +0 -0
  70. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/model/config.py +0 -0
  71. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/model/mixins/logger.py +0 -0
  72. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/nn/__init__.py +0 -0
  73. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/nn/mlp.py +0 -0
  74. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/nn/module_dict.py +0 -0
  75. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/nn/module_list.py +0 -0
  76. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/nn/nonlinearity.py +0 -0
  77. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/optimizer.py +0 -0
  78. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/profiler/__init__.py +0 -0
  79. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/profiler/_base.py +0 -0
  80. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/profiler/advanced.py +0 -0
  81. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/profiler/pytorch.py +0 -0
  82. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/profiler/simple.py +0 -0
  83. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/runner.py +0 -0
  84. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/scripts/find_packages.py +0 -0
  85. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/trainer/__init__.py +0 -0
  86. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/trainer/_config.py +0 -0
  87. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/trainer/_runtime_callback.py +0 -0
  88. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/trainer/checkpoint_connector.py +0 -0
  89. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/trainer/signal_connector.py +0 -0
  90. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/trainer/trainer.py +0 -0
  91. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/_environment_info.py +0 -0
  92. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/_useful_types.py +0 -0
  93. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/config/__init__.py +0 -0
  94. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/config/duration.py +0 -0
  95. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/environment.py +0 -0
  96. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/path.py +0 -0
  97. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/seed.py +0 -0
  98. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/slurm.py +0 -0
  99. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/src/nshtrainer/util/typed.py +0 -0
  100. {nshtrainer-0.33.0 → nshtrainer-0.33.1}/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.33.0
3
+ Version: 0.33.1
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.33.0"
3
+ version = "0.33.1"
4
4
  description = ""
5
5
  authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
6
6
  readme = "README.md"
@@ -53,12 +53,48 @@ from nshtrainer.callbacks.throughput_monitor import (
53
53
  )
54
54
  from nshtrainer.callbacks.timer import EpochTimerConfig as EpochTimerConfig
55
55
  from nshtrainer.callbacks.wandb_watch import WandbWatchConfig as WandbWatchConfig
56
+ from nshtrainer.config import UUID1 as UUID1
57
+ from nshtrainer.config import UUID3 as UUID3
58
+ from nshtrainer.config import UUID4 as UUID4
59
+ from nshtrainer.config import UUID5 as UUID5
60
+ from nshtrainer.config import AmqpDsn as AmqpDsn
61
+ from nshtrainer.config import AnyHttpUrl as AnyHttpUrl
62
+ from nshtrainer.config import AnyWebsocketUrl as AnyWebsocketUrl
63
+ from nshtrainer.config import Base64Bytes as Base64Bytes
64
+ from nshtrainer.config import Base64Str as Base64Str
65
+ from nshtrainer.config import Base64UrlBytes as Base64UrlBytes
66
+ from nshtrainer.config import Base64UrlStr as Base64UrlStr
67
+ from nshtrainer.config import ClickHouseDsn as ClickHouseDsn
68
+ from nshtrainer.config import CockroachDsn as CockroachDsn
69
+ from nshtrainer.config import DirectoryPath as DirectoryPath
70
+ from nshtrainer.config import FilePath as FilePath
71
+ from nshtrainer.config import FileUrl as FileUrl
72
+ from nshtrainer.config import FiniteFloat as FiniteFloat
73
+ from nshtrainer.config import FtpUrl as FtpUrl
74
+ from nshtrainer.config import HttpUrl as HttpUrl
75
+ from nshtrainer.config import KafkaDsn as KafkaDsn
76
+ from nshtrainer.config import MariaDBDsn as MariaDBDsn
77
+ from nshtrainer.config import MongoDsn as MongoDsn
78
+ from nshtrainer.config import MySQLDsn as MySQLDsn
79
+ from nshtrainer.config import NatsDsn as NatsDsn
80
+ from nshtrainer.config import NewPath as NewPath
81
+ from nshtrainer.config import OnErrorOmit as OnErrorOmit
82
+ from nshtrainer.config import PostgresDsn as PostgresDsn
83
+ from nshtrainer.config import RedisDsn as RedisDsn
84
+ from nshtrainer.config import SnowflakeDsn as SnowflakeDsn
85
+ from nshtrainer.config import StrictBool as StrictBool
86
+ from nshtrainer.config import StrictBytes as StrictBytes
87
+ from nshtrainer.config import StrictFloat as StrictFloat
88
+ from nshtrainer.config import StrictInt as StrictInt
89
+ from nshtrainer.config import StrictStr as StrictStr
90
+ from nshtrainer.config import WebsocketUrl as WebsocketUrl
56
91
  from nshtrainer.loggers._base import BaseLoggerConfig as BaseLoggerConfig
57
92
  from nshtrainer.loggers.csv import CSVLoggerConfig as CSVLoggerConfig
58
93
  from nshtrainer.loggers.tensorboard import (
59
94
  TensorboardLoggerConfig as TensorboardLoggerConfig,
60
95
  )
61
96
  from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
97
+ from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
62
98
  from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase
63
99
  from nshtrainer.lr_scheduler.linear_warmup_cosine import (
64
100
  DurationConfig as DurationConfig,
@@ -103,6 +139,7 @@ from nshtrainer.nn.nonlinearity import (
103
139
  )
104
140
  from nshtrainer.nn.nonlinearity import TanhNonlinearityConfig as TanhNonlinearityConfig
105
141
  from nshtrainer.optimizer import AdamWConfig as AdamWConfig
142
+ from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
106
143
  from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
107
144
  from nshtrainer.profiler._base import BaseProfilerConfig as BaseProfilerConfig
108
145
  from nshtrainer.profiler.advanced import (
File without changes