nshtrainer 1.0.0b21__py3-none-any.whl → 1.0.0b22__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 (4) hide show
  1. nshtrainer/trainer/trainer.py +9 -0
  2. {nshtrainer-1.0.0b21.dist-info → nshtrainer-1.0.0b22.dist-info}/METADATA +1 -1
  3. {nshtrainer-1.0.0b21.dist-info → nshtrainer-1.0.0b22.dist-info}/RECORD +4 -4
  4. {nshtrainer-1.0.0b21.dist-info → nshtrainer-1.0.0b22.dist-info}/WHEEL +0 -0
nshtrainer/trainer/trainer.py CHANGED
@@ -316,8 +316,17 @@ class Trainer(LightningTrainer):
316
316
  self._hparams = hparams
317
317
  self.debug = self.hparams.debug
318
318
 
319
+ experimental_profiler = None
320
+ if hparams.experimental_barebones_profiler_enabled:
321
+ experimental_profiler = kwargs.pop("profiler", None)
319
322
  super().__init__(**kwargs)
320
323
 
324
+ # Set up the profiler again
325
+ if experimental_profiler is not None:
326
+ from lightning.pytorch.trainer import setup
327
+
328
+ setup._init_profiler(self, experimental_profiler)
329
+
321
330
  # Add our own start time callback to measure the start time.
322
331
  self.callbacks.append(RuntimeTrackerCallback())
323
332
 
{nshtrainer-1.0.0b21.dist-info → nshtrainer-1.0.0b22.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 1.0.0b21
3
+ Version: 1.0.0b22
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-1.0.0b21.dist-info → nshtrainer-1.0.0b22.dist-info}/RECORD RENAMED
@@ -122,7 +122,7 @@ nshtrainer/trainer/__init__.py,sha256=MmoydVS6aYeav7zgDAUHxAQrV_PMQsbnZTCuPnLH9W
122
122
  nshtrainer/trainer/_config.py,sha256=xd80gDFSj5QvRIZ_2t15PN397akpXX8VUazeivS8kV0,34112
123
123
  nshtrainer/trainer/_runtime_callback.py,sha256=T3epaj1YeIN0R8CS2cg5HNJIB21TyaD_PVNNOPJ6nJs,4200
124
124
  nshtrainer/trainer/signal_connector.py,sha256=YMJf6vTnW0JcnBkuYikm9x_9XscaokrCEzCn4THOGao,10776
125
- nshtrainer/trainer/trainer.py,sha256=WydBYDuRpGFGkl180Wss6ADIu-fo_kQeDi-0LI64OyM,19770
125
+ nshtrainer/trainer/trainer.py,sha256=SgAjALn8LsRWmvBPvHJOP0IKDZOKKl3lwfrC1q4J2EA,20134
126
126
  nshtrainer/util/_environment_info.py,sha256=MT8mBe6ZolRfKiwU-les1P-lPNPqXpHQcfADrh_A3uY,24629
127
127
  nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
128
128
  nshtrainer/util/bf16.py,sha256=9QhHZCkYSfYpIcxwAMoXyuh2yTSHBzT-EdLQB297jEs,762
@@ -135,6 +135,6 @@ nshtrainer/util/seed.py,sha256=diMV8iwBKN7Xxt5pELmui-gyqyT80_CZzomrWhNss0k,316
135
135
  nshtrainer/util/slurm.py,sha256=HflkP5iI_r4UHMyPjw9R4dD5AHsJUpcfJw5PLvGYBRM,1603
136
136
  nshtrainer/util/typed.py,sha256=Xt5fUU6zwLKSTLUdenovnKK0N8qUq89Kddz2_XeykVQ,164
137
137
  nshtrainer/util/typing_utils.py,sha256=MjY-CUX9R5Tzat-BlFnQjwl1PQ_W2yZQoXhkYHlJ_VA,442
138
- nshtrainer-1.0.0b21.dist-info/METADATA,sha256=PwOjPxm1i_uXzfl6QBdjHcNnbwtbbXwgoIFxkBg7PFU,937
139
- nshtrainer-1.0.0b21.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
140
- nshtrainer-1.0.0b21.dist-info/RECORD,,
138
+ nshtrainer-1.0.0b22.dist-info/METADATA,sha256=AkEEGt5cpSMgefZzzZWfTEu6DztnaodTTg9aOtiWQc8,937
139
+ nshtrainer-1.0.0b22.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
140
+ nshtrainer-1.0.0b22.dist-info/RECORD,,