nshtrainer 0.8.2__py3-none-any.whl → 0.8.4__py3-none-any.whl

nshtrainer/__init__.py

@@ -1,42 +1,14 @@
 from . import _experimental as _experimental
 from . import callbacks as callbacks
+from . import data as data
 from . import lr_scheduler as lr_scheduler
 from . import model as model
 from . import nn as nn
 from . import optimizer as optimizer
-from .data import dataset_transform as dataset_transform
-from .lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
-from .model import ActSaveConfig as ActSaveConfig
 from .model import Base as Base
 from .model import BaseConfig as BaseConfig
-from .model import BaseLoggerConfig as BaseLoggerConfig
-from .model import BaseProfilerConfig as BaseProfilerConfig
-from .model import CheckpointLoadingConfig as CheckpointLoadingConfig
-from .model import CheckpointSavingConfig as CheckpointSavingConfig
 from .model import ConfigList as ConfigList
-from .model import DirectoryConfig as DirectoryConfig
-from .model import (
-    EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
-)
-from .model import EnvironmentConfig as EnvironmentConfig
-from .model import (
-    EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
-)
-from .model import (
-    EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
-)
-from .model import GradientClippingConfig as GradientClippingConfig
 from .model import LightningModuleBase as LightningModuleBase
-from .model import LoggingConfig as LoggingConfig
 from .model import MetricConfig as MetricConfig
-from .model import OptimizationConfig as OptimizationConfig
-from .model import PrimaryMetricConfig as PrimaryMetricConfig
-from .model import ReproducibilityConfig as ReproducibilityConfig
-from .model import SanityCheckingConfig as SanityCheckingConfig
-from .model import TrainerConfig as TrainerConfig
-from .model import WandbWatchConfig as WandbWatchConfig
-from .nn import TypedModuleDict as TypedModuleDict
-from .nn import TypedModuleList as TypedModuleList
-from .optimizer import OptimizerConfig as OptimizerConfig
 from .runner import Runner as Runner
 from .trainer import Trainer as Trainer

nshtrainer/ll/__init__.py

@@ -1,6 +1,7 @@
 from . import _experimental as _experimental
 from . import actsave as actsave
 from . import callbacks as callbacks
+from . import data as data
 from . import lr_scheduler as lr_scheduler
 from . import model as model
 from . import nn as nn

nshtrainer/ll/_experimental.py

@@ -0,0 +1 @@
+from nshtrainer._experimental import *  # noqa: F403

nshtrainer/ll/snapshot.py

@@ -0,0 +1 @@
+from nshrunner.snapshot import *  # type: ignore # noqa: F403

{nshtrainer-0.8.2.dist-info → nshtrainer-0.8.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 0.8.2
+Version: 0.8.4
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-0.8.2.dist-info → nshtrainer-0.8.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-nshtrainer/__init__.py,sha256=gWRJYlPoS0Xlyy1I0hob2zKxuqDB2fO4cfAryb--Prk,1992
+nshtrainer/__init__.py,sha256=nbZHdfTk0oWqsJgrSzdgk2DSf4CGhdZn79esoJGauO8,548
 nshtrainer/_experimental/__init__.py,sha256=2tQIcrWT8U8no_AeBTYnozaTmxN40kuAJdGQ4b-PoWM,120
 nshtrainer/_experimental/flops/__init__.py,sha256=edo9Ez3LlrnxkNRX9W6YBhPkRPKYGLpkpnl5gx7sEX8,1550
 nshtrainer/_experimental/flops/flop_counter.py,sha256=-sL0Fy6poXa__hyzUMdZScjPULp4coQELQpPU6p6dXU,25736
@@ -24,7 +24,8 @@ nshtrainer/callbacks/wandb_watch.py,sha256=pUpMsNxd03ex1rzOmFw2HzGOXjnQGaH84m8cc
 nshtrainer/data/__init__.py,sha256=7mk1tr7SWUZ7ySbsf0y0ZPszk7u4QznPhQ-7wnpH9ec,149
 nshtrainer/data/balanced_batch_sampler.py,sha256=bcJBcQjh1hB1yKF_xSlT9AtEWv0BJjYc1CuH2BF-ea8,4392
 nshtrainer/data/transform.py,sha256=JeGxvytQly8hougrsdMmKG8gJ6qvFPDglJCO4Tp6STk,1795
-nshtrainer/ll/__init__.py,sha256=qraZm_bAvoLR69w5iiAic9-TT4sR0g_iKnttiXbb5y0,2658
+nshtrainer/ll/__init__.py,sha256=nxYPtoFOFAvzkD6O3EIuwCiRi_LedYa_EH-RIfDG91s,2685
+nshtrainer/ll/_experimental.py,sha256=oBQCKOEVYoxuUU9eLb-Fg2B2mzZD7SA0zfAO6lmWZ88,53
 nshtrainer/ll/actsave.py,sha256=QJ7yJIqvabpZzumX7PLPzkh6dfqY-zxiEdzv48VtZEY,123
 nshtrainer/ll/callbacks.py,sha256=AxyUmc8aGRSjx6WwwgXYCmdJ73rwLuEAEH0AGRosojQ,49
 nshtrainer/ll/config.py,sha256=fKumJf42HY2FITX1QUM1OTXkYD6U2np2ciyd4PFRPZ8,145
@@ -35,6 +36,7 @@ nshtrainer/ll/model.py,sha256=6I9gQjEFT2Veer-UmcPy05Pt3mnvaxYX1b3sOdaF96A,45
 nshtrainer/ll/nn.py,sha256=8qiRDFwojIxkB7-LtNWk4mLL2tJbaskHYofDsOIHiNg,42
 nshtrainer/ll/optimizer.py,sha256=3T-VZtT73jVvwCNJGDjgGEbzs-1LFTzMQH-SB_58mSo,49
 nshtrainer/ll/runner.py,sha256=B0m5VEhNKIjF1aFmqPkonkQxDoRL2jeHZGsV3zwhSVE,117
+nshtrainer/ll/snapshot.py,sha256=UB3AjU6kjD6UkKcJg0y2m9rHQbjZ78NaRuTdk7TdxUI,62
 nshtrainer/ll/snoop.py,sha256=hG9VCdm8mIZytHLZgUKRSoWqg55rVvUpBAH-OSiwgvI,36
 nshtrainer/ll/trainer.py,sha256=hkn2xPtrSPQ7LqQhbyAKuMfNyHdhqB9bDPvgRCK8oJM,47
 nshtrainer/ll/typecheck.py,sha256=ryV1Tzcf7hJ4I19H1oQVkikU9spmRk8jyIKQZ5UF7pQ,62
@@ -70,6 +72,6 @@ nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
 nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
 nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
 nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
-nshtrainer-0.8.2.dist-info/METADATA,sha256=R2vJa03efqHrq_Jym3vmEZtt3muHmNCDNSYRVrkxlis,647
-nshtrainer-0.8.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-0.8.2.dist-info/RECORD,,
+nshtrainer-0.8.4.dist-info/METADATA,sha256=PS0Z7icSp22Ok5Bs6SnOc0PEBqg42kjXTnpW5jqQaxw,647
+nshtrainer-0.8.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-0.8.4.dist-info/RECORD,,