nshtrainer 0.10.0__tar.gz → 0.10.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 (84) hide show
  1. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/PKG-INFO +1 -1
  2. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/pyproject.toml +1 -1
  3. nshtrainer-0.10.0/src/nshtrainer/trainer/_checkpoint_resolver.py → nshtrainer-0.10.1/src/nshtrainer/_checkpoint/loader.py +1 -1
  4. nshtrainer-0.10.0/src/nshtrainer/trainer/_checkpoint_metadata.py → nshtrainer-0.10.1/src/nshtrainer/_checkpoint/metadata.py +1 -1
  5. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/config.py +1 -1
  6. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/trainer/checkpoint_connector.py +1 -1
  7. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/trainer/trainer.py +1 -1
  8. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/README.md +0 -0
  9. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/__init__.py +0 -0
  10. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/_experimental/__init__.py +0 -0
  11. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/_experimental/flops/__init__.py +0 -0
  12. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/_experimental/flops/flop_counter.py +0 -0
  13. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/_experimental/flops/module_tracker.py +0 -0
  14. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/__init__.py +0 -0
  15. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/_throughput_monitor_callback.py +0 -0
  16. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/actsave.py +0 -0
  17. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/base.py +0 -0
  18. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/early_stopping.py +0 -0
  19. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/ema.py +0 -0
  20. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/finite_checks.py +0 -0
  21. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/gradient_skipping.py +0 -0
  22. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/interval.py +0 -0
  23. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/latest_epoch_checkpoint.py +0 -0
  24. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/log_epoch.py +0 -0
  25. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/model_checkpoint.py +0 -0
  26. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/norm_logging.py +0 -0
  27. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/on_exception_checkpoint.py +0 -0
  28. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/print_table.py +0 -0
  29. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/throughput_monitor.py +0 -0
  30. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/timer.py +0 -0
  31. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/callbacks/wandb_watch.py +0 -0
  32. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/data/__init__.py +0 -0
  33. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/data/balanced_batch_sampler.py +0 -0
  34. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/data/transform.py +0 -0
  35. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/__init__.py +0 -0
  36. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/_experimental.py +0 -0
  37. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/actsave.py +0 -0
  38. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/callbacks.py +0 -0
  39. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/config.py +0 -0
  40. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/data.py +0 -0
  41. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/log.py +0 -0
  42. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/lr_scheduler.py +0 -0
  43. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/model.py +0 -0
  44. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/nn.py +0 -0
  45. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/optimizer.py +0 -0
  46. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/runner.py +0 -0
  47. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/snapshot.py +0 -0
  48. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/snoop.py +0 -0
  49. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/trainer.py +0 -0
  50. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/typecheck.py +0 -0
  51. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/ll/util.py +0 -0
  52. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/lr_scheduler/__init__.py +0 -0
  53. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/lr_scheduler/_base.py +0 -0
  54. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/lr_scheduler/linear_warmup_cosine.py +0 -0
  55. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/lr_scheduler/reduce_lr_on_plateau.py +0 -0
  56. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/metrics/__init__.py +0 -0
  57. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/metrics/_config.py +0 -0
  58. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/__init__.py +0 -0
  59. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/_environment.py +0 -0
  60. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/base.py +0 -0
  61. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/modules/callback.py +0 -0
  62. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/modules/debug.py +0 -0
  63. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/modules/distributed.py +0 -0
  64. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/modules/logger.py +0 -0
  65. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/modules/profiler.py +0 -0
  66. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/modules/rlp_sanity_checks.py +0 -0
  67. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/modules/shared_parameters.py +0 -0
  68. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/nn/__init__.py +0 -0
  69. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/nn/mlp.py +0 -0
  70. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/nn/module_dict.py +0 -0
  71. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/nn/module_list.py +0 -0
  72. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/nn/nonlinearity.py +0 -0
  73. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/optimizer.py +0 -0
  74. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/runner.py +0 -0
  75. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/scripts/check_env.py +0 -0
  76. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/scripts/find_packages.py +0 -0
  77. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/trainer/__init__.py +0 -0
  78. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/trainer/_runtime_callback.py +0 -0
  79. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/trainer/signal_connector.py +0 -0
  80. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/util/environment.py +0 -0
  81. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/util/seed.py +0 -0
  82. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/util/slurm.py +0 -0
  83. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/util/typed.py +0 -0
  84. {nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/util/typing_utils.py +0 -0
{nshtrainer-0.10.0 → nshtrainer-0.10.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.10.0
3
+ Version: 0.10.1
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.10.0 → nshtrainer-0.10.1}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "nshtrainer"
3
- version = "0.10.0"
3
+ version = "0.10.1"
4
4
  description = ""
5
5
  authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
6
6
  readme = "README.md"
nshtrainer-0.10.0/src/nshtrainer/trainer/_checkpoint_resolver.py → nshtrainer-0.10.1/src/nshtrainer/_checkpoint/loader.py RENAMED
@@ -10,7 +10,7 @@ from lightning.pytorch.trainer.states import TrainerFn
10
10
  from typing_extensions import assert_never
11
11
 
12
12
  from ..metrics._config import MetricConfig
13
- from ._checkpoint_metadata import METADATA_PATH_SUFFIX, CheckpointMetadata
13
+ from .metadata import METADATA_PATH_SUFFIX, CheckpointMetadata
14
14
 
15
15
  if TYPE_CHECKING:
16
16
  from ..model.config import BaseConfig
nshtrainer-0.10.0/src/nshtrainer/trainer/_checkpoint_metadata.py → nshtrainer-0.10.1/src/nshtrainer/_checkpoint/metadata.py RENAMED
@@ -12,7 +12,7 @@ from ..model._environment import EnvironmentConfig
12
12
 
13
13
  if TYPE_CHECKING:
14
14
  from ..model import BaseConfig, LightningModuleBase
15
- from .trainer import Trainer
15
+ from ..trainer.trainer import Trainer
16
16
 
17
17
  log = logging.getLogger(__name__)
18
18
 
{nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/model/config.py RENAMED
@@ -34,6 +34,7 @@ from lightning.pytorch.strategies.strategy import Strategy
34
34
  from pydantic import DirectoryPath
35
35
  from typing_extensions import Self, TypedDict, TypeVar, override
36
36
 
37
+ from .._checkpoint.loader import CheckpointLoadingConfig
37
38
  from ..callbacks import (
38
39
  CallbackConfig,
39
40
  LatestEpochCheckpointCallbackConfig,
@@ -43,7 +44,6 @@ from ..callbacks import (
43
44
  )
44
45
  from ..callbacks.base import CallbackConfigBase
45
46
  from ..metrics import MetricConfig
46
- from ..trainer._checkpoint_resolver import CheckpointLoadingConfig
47
47
  from ._environment import EnvironmentConfig
48
48
 
49
49
  log = getLogger(__name__)
{nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/trainer/checkpoint_connector.py RENAMED
@@ -8,7 +8,7 @@ from lightning.pytorch.trainer.connectors.checkpoint_connector import (
8
8
  from lightning.pytorch.trainer.states import TrainerFn
9
9
  from typing_extensions import override
10
10
 
11
- from ._checkpoint_resolver import CheckpointLoadingConfig, _resolve_checkpoint
11
+ from .._checkpoint.loader import CheckpointLoadingConfig, _resolve_checkpoint
12
12
 
13
13
  if TYPE_CHECKING:
14
14
  from ..model.config import BaseConfig
{nshtrainer-0.10.0 → nshtrainer-0.10.1}/src/nshtrainer/trainer/trainer.py RENAMED
@@ -17,6 +17,7 @@ from lightning.pytorch.trainer.states import TrainerFn
17
17
  from lightning.pytorch.utilities.types import _EVALUATE_OUTPUT, _PREDICT_OUTPUT
18
18
  from typing_extensions import Unpack, assert_never, override
19
19
 
20
+ from .._checkpoint.metadata import _write_checkpoint_metadata
20
21
  from ..callbacks.base import resolve_all_callbacks
21
22
  from ..model.config import (
22
23
  AcceleratorConfigProtocol,
@@ -25,7 +26,6 @@ from ..model.config import (
25
26
  LightningTrainerKwargs,
26
27
  StrategyConfigProtocol,
27
28
  )
28
- from ._checkpoint_metadata import _write_checkpoint_metadata
29
29
  from ._runtime_callback import RuntimeTrackerCallback, Stage
30
30
  from .signal_connector import _SignalConnector
31
31
 
{nshtrainer-0.10.0 → nshtrainer-0.10.1}/README.md RENAMED
File without changes