nshtrainer 0.1.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

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 (6) hide show
  1. nshtrainer/__init__.py +1 -1
  2. nshtrainer/config.py +4 -0
  3. nshtrainer/runner.py +3 -3
  4. {nshtrainer-0.1.1.dist-info → nshtrainer-0.2.0.dist-info}/METADATA +2 -2
  5. {nshtrainer-0.1.1.dist-info → nshtrainer-0.2.0.dist-info}/RECORD +6 -5
  6. {nshtrainer-0.1.1.dist-info → nshtrainer-0.2.0.dist-info}/WHEEL +0 -0
nshtrainer/__init__.py CHANGED
@@ -1,11 +1,11 @@
1
1
  from . import _experimental as _experimental
2
2
  from . import actsave as actsave
3
3
  from . import callbacks as callbacks
4
+ from . import config as config
4
5
  from . import lr_scheduler as lr_scheduler
5
6
  from . import model as model
6
7
  from . import nn as nn
7
8
  from . import optimizer as optimizer
8
- from . import snapshot as snapshot
9
9
  from . import typecheck as typecheck
10
10
  from ._snoop import snoop as snoop
11
11
  from .actsave import ActLoad as ActLoad
nshtrainer/config.py ADDED
@@ -0,0 +1,4 @@
1
+ from nshconfig import * # type: ignore # noqa: F403
2
+ from nshconfig import Config
3
+
4
+ TypedConfig = Config
nshtrainer/runner.py CHANGED
@@ -7,13 +7,13 @@ from typing_extensions import TypeVar, TypeVarTuple, Unpack, override
7
7
  from .model.config import BaseConfig
8
8
 
9
9
  TConfig = TypeVar("TConfig", bound=BaseConfig, infer_variance=True)
10
- TArguments = TypeVarTuple("TArguments")
10
+ TArguments = TypeVarTuple("TArguments", default=Unpack[tuple[()]])
11
11
  TReturn = TypeVar("TReturn", infer_variance=True)
12
12
 
13
13
 
14
14
  class Runner(
15
- _Runner[Unpack[tuple[TConfig, Unpack[TArguments]]], TReturn],
16
- Generic[TConfig, Unpack[TArguments], TReturn],
15
+ _Runner[TReturn, TConfig, Unpack[TArguments]],
16
+ Generic[TReturn, TConfig, Unpack[TArguments]],
17
17
  ):
18
18
  @override
19
19
  @classmethod
{nshtrainer-0.1.1.dist-info → nshtrainer-0.2.0.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.1.1
3
+ Version: 0.2.0
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
@@ -15,7 +15,7 @@ Requires-Dist: lightning
15
15
  Requires-Dist: lovely-numpy (>=0.2.13,<0.3.0)
16
16
  Requires-Dist: lovely-tensors (>=0.1.16,<0.2.0)
17
17
  Requires-Dist: nshconfig (>=0.2.0,<0.3.0)
18
- Requires-Dist: nshrunner (>=0.5.3,<0.6.0)
18
+ Requires-Dist: nshrunner (>=0.5.4,<0.6.0)
19
19
  Requires-Dist: numpy
20
20
  Requires-Dist: pysnooper
21
21
  Requires-Dist: pytorch-lightning
{nshtrainer-0.1.1.dist-info → nshtrainer-0.2.0.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- nshtrainer/__init__.py,sha256=OHbxLxVvFGW--ecuIGqkoylSVHFS4x4F1-oeuENH-Do,2212
1
+ nshtrainer/__init__.py,sha256=_r7kBmgGSLVfActlqQeupNolrmBu45xUuSS8odt3HL8,2208
2
2
  nshtrainer/_experimental/__init__.py,sha256=2tQIcrWT8U8no_AeBTYnozaTmxN40kuAJdGQ4b-PoWM,120
3
3
  nshtrainer/_experimental/flops/__init__.py,sha256=edo9Ez3LlrnxkNRX9W6YBhPkRPKYGLpkpnl5gx7sEX8,1550
4
4
  nshtrainer/_experimental/flops/flop_counter.py,sha256=-sL0Fy6poXa__hyzUMdZScjPULp4coQELQpPU6p6dXU,25736
@@ -24,6 +24,7 @@ nshtrainer/callbacks/print_table.py,sha256=FcA-CBWwMf9c1NNRinvYpZC400RNQxuP28bJf
24
24
  nshtrainer/callbacks/throughput_monitor.py,sha256=YQLdpX3LGybIiD814yT9yCCVSEXRWf8WwsvVaN5aDBE,1848
25
25
  nshtrainer/callbacks/timer.py,sha256=sDXPPcdDKu5xnuK_bjr8plIq9MBuluNJ42Mt9LvPZzc,4610
26
26
  nshtrainer/callbacks/wandb_watch.py,sha256=pUpMsNxd03ex1rzOmFw2HzGOXjnQGaH84m8cc2dXo4g,2937
27
+ nshtrainer/config.py,sha256=IXOAl_JWFNX9kPTo_iw4Nc3qXqkKrbA6-ZrvTAjqu6A,104
27
28
  nshtrainer/data/__init__.py,sha256=7mk1tr7SWUZ7ySbsf0y0ZPszk7u4QznPhQ-7wnpH9ec,149
28
29
  nshtrainer/data/balanced_batch_sampler.py,sha256=bcJBcQjh1hB1yKF_xSlT9AtEWv0BJjYc1CuH2BF-ea8,4392
29
30
  nshtrainer/data/transform.py,sha256=JeGxvytQly8hougrsdMmKG8gJ6qvFPDglJCO4Tp6STk,1795
@@ -47,7 +48,7 @@ nshtrainer/nn/module_dict.py,sha256=NOY0B6WDTnktyWH4GthsprMQo0bpehC-hCq9SfD8paE,
47
48
  nshtrainer/nn/module_list.py,sha256=fb2u5Rqdjff8Pekyr9hkCPkBorQ-fldzzFAjsgWAm30,1719
48
49
  nshtrainer/nn/nonlinearity.py,sha256=owtU4kh4G98psD0axOJWVfBhm-OtJVgFM-TXSHmbNPU,3625
49
50
  nshtrainer/optimizer.py,sha256=kuJEA1pvB3y1FcsfhAoOJujVqEZqFHlmYO8GW6JeA1g,1527
50
- nshtrainer/runner.py,sha256=af_EGnQTSvUgwnVhhytvY3V7o_Xg-xx-sLb8K2Szb1E,979
51
+ nshtrainer/runner.py,sha256=vyHr0EZ0PBOWZh09BtOOxio-FRQZFbVoL4cdBlI97vY,991
51
52
  nshtrainer/scripts/check_env.py,sha256=IMl6dSqsLYppI0XuCsVq8lK4bYqXwY9KHJkzsShz4Kg,806
52
53
  nshtrainer/scripts/find_packages.py,sha256=FbdlfmAefttFSMfaT0A46a-oHLP_ioaQKihwBfBeWeA,1467
53
54
  nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
@@ -59,6 +60,6 @@ nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
59
60
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
60
61
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
61
62
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
62
- nshtrainer-0.1.1.dist-info/METADATA,sha256=32iVLvdJh6OJQyD-_7NDO6IYqfHPSflDznYfYaCo8-c,882
63
- nshtrainer-0.1.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
64
- nshtrainer-0.1.1.dist-info/RECORD,,
63
+ nshtrainer-0.2.0.dist-info/METADATA,sha256=cwb3IbKGyJ9HbNSvsORYhCiI61nrDMb1dVm5nE1q_XA,882
64
+ nshtrainer-0.2.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
65
+ nshtrainer-0.2.0.dist-info/RECORD,,