torchx-nightly 2024.6.28__py3-none-any.whl → 2024.7.3__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.
Potentially problematic release.
This version of torchx-nightly might be problematic. Click here for more details.
- torchx/examples/apps/lightning/profiler.py +5 -4
- {torchx_nightly-2024.6.28.dist-info → torchx_nightly-2024.7.3.dist-info}/METADATA +3 -5
- {torchx_nightly-2024.6.28.dist-info → torchx_nightly-2024.7.3.dist-info}/RECORD +7 -7
- {torchx_nightly-2024.6.28.dist-info → torchx_nightly-2024.7.3.dist-info}/entry_points.txt +0 -1
- {torchx_nightly-2024.6.28.dist-info → torchx_nightly-2024.7.3.dist-info}/LICENSE +0 -0
- {torchx_nightly-2024.6.28.dist-info → torchx_nightly-2024.7.3.dist-info}/WHEEL +0 -0
- {torchx_nightly-2024.6.28.dist-info → torchx_nightly-2024.7.3.dist-info}/top_level.txt +0 -0
|
@@ -19,18 +19,19 @@ output is used for HPO optimization with Ax.
|
|
|
19
19
|
import time
|
|
20
20
|
from typing import Dict
|
|
21
21
|
|
|
22
|
-
from pytorch_lightning.loggers.
|
|
23
|
-
from pytorch_lightning.profiler.base import BaseProfiler
|
|
22
|
+
from pytorch_lightning.loggers.logger import Logger
|
|
24
23
|
|
|
24
|
+
from pytorch_lightning.profilers.profiler import Profiler
|
|
25
25
|
|
|
26
|
-
|
|
26
|
+
|
|
27
|
+
class SimpleLoggingProfiler(Profiler):
|
|
27
28
|
"""
|
|
28
29
|
This profiler records the duration of actions (in seconds) and reports the
|
|
29
30
|
mean duration of each action to the specified logger. Reported metrics are
|
|
30
31
|
in the format `duration_<event>`.
|
|
31
32
|
"""
|
|
32
33
|
|
|
33
|
-
def __init__(self, logger:
|
|
34
|
+
def __init__(self, logger: Logger) -> None:
|
|
34
35
|
super().__init__()
|
|
35
36
|
|
|
36
37
|
self.current_actions: Dict[str, float] = {}
|
|
@@ -1,13 +1,12 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: torchx-nightly
|
|
3
|
-
Version: 2024.
|
|
3
|
+
Version: 2024.7.3
|
|
4
4
|
Summary: TorchX SDK and Components
|
|
5
5
|
Home-page: https://github.com/pytorch/torchx
|
|
6
6
|
Author: TorchX Devs
|
|
7
7
|
Author-email: torchx@fb.com
|
|
8
8
|
License: BSD-3
|
|
9
9
|
Keywords: pytorch,machine learning
|
|
10
|
-
Platform: UNKNOWN
|
|
11
10
|
Classifier: Development Status :: 4 - Beta
|
|
12
11
|
Classifier: Intended Audience :: Developers
|
|
13
12
|
Classifier: Intended Audience :: Science/Research
|
|
@@ -52,7 +51,8 @@ Requires-Dist: pyre-extensions ; extra == 'dev'
|
|
|
52
51
|
Requires-Dist: pyre-check ; extra == 'dev'
|
|
53
52
|
Requires-Dist: pytest ; extra == 'dev'
|
|
54
53
|
Requires-Dist: pytest-cov ; extra == 'dev'
|
|
55
|
-
Requires-Dist: pytorch-lightning ==
|
|
54
|
+
Requires-Dist: pytorch-lightning ==2.3.1 ; extra == 'dev'
|
|
55
|
+
Requires-Dist: tensorboard ==2.14.0 ; extra == 'dev'
|
|
56
56
|
Requires-Dist: sagemaker >=2.149.0 ; extra == 'dev'
|
|
57
57
|
Requires-Dist: torch-model-archiver >=0.4.2 ; extra == 'dev'
|
|
58
58
|
Requires-Dist: torch ==2.2.1 ; extra == 'dev'
|
|
@@ -180,5 +180,3 @@ We welcome PRs! See the [CONTRIBUTING](https://github.com/pytorch/torchx/blob/ma
|
|
|
180
180
|
## License
|
|
181
181
|
|
|
182
182
|
TorchX is BSD licensed, as found in the [LICENSE](https://github.com/pytorch/torchx/blob/main/LICENSE) file.
|
|
183
|
-
|
|
184
|
-
|
|
@@ -44,7 +44,7 @@ torchx/examples/apps/lightning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
|
|
|
44
44
|
torchx/examples/apps/lightning/data.py,sha256=Sp9mv29FlCtMdMeSKOrHan_QHK-h_crCIkCXopb20kQ,6598
|
|
45
45
|
torchx/examples/apps/lightning/interpret.py,sha256=Hd3kE5a6FyhxCmJBfTzb4Tlj518zhX8V0XvZfzu4nqE,5256
|
|
46
46
|
torchx/examples/apps/lightning/model.py,sha256=ppj8pYkJ1Zj4kZX6JXULGgx3sYVfNIQ24OuDYXc5lYo,3947
|
|
47
|
-
torchx/examples/apps/lightning/profiler.py,sha256=
|
|
47
|
+
torchx/examples/apps/lightning/profiler.py,sha256=SSSihnwjeUTkBoz0E3qn1b-wbkfUIowscx2ND_37zyw,1915
|
|
48
48
|
torchx/examples/apps/lightning/train.py,sha256=MGSapykGv-m4nl0WyRC-yzhPWEI4bHQ5WKrniR6czQk,6099
|
|
49
49
|
torchx/examples/pipelines/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
50
50
|
torchx/examples/pipelines/kfp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
@@ -113,9 +113,9 @@ torchx/workspace/__init__.py,sha256=FqN8AN4VhR1C_SBY10MggQvNZmyanbbuPuE-JCjkyUY,
|
|
|
113
113
|
torchx/workspace/api.py,sha256=1heBmPgB-W5Zf9gwViM7NrqvHpZlVYeMN7jpY8Qkytc,5479
|
|
114
114
|
torchx/workspace/dir_workspace.py,sha256=npNW_IjUZm_yS5r-8hrRkH46ndDd9a_eApT64m1S1T4,2268
|
|
115
115
|
torchx/workspace/docker_workspace.py,sha256=PFu2KQNVC-0p2aKJ-W_BKA9ZOmXdCY2ABEkCExp3udQ,10269
|
|
116
|
-
torchx_nightly-2024.
|
|
117
|
-
torchx_nightly-2024.
|
|
118
|
-
torchx_nightly-2024.
|
|
119
|
-
torchx_nightly-2024.
|
|
120
|
-
torchx_nightly-2024.
|
|
121
|
-
torchx_nightly-2024.
|
|
116
|
+
torchx_nightly-2024.7.3.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
|
|
117
|
+
torchx_nightly-2024.7.3.dist-info/METADATA,sha256=_s1p6DZ_v6Rg9pBbS5No2a9yYGRX4k0nWXqLG4evubE,6215
|
|
118
|
+
torchx_nightly-2024.7.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
|
|
119
|
+
torchx_nightly-2024.7.3.dist-info/entry_points.txt,sha256=T328AMXeKI3JZnnxfkEew2ZcMN1oQDtkXjMz7lkV-P4,169
|
|
120
|
+
torchx_nightly-2024.7.3.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
|
|
121
|
+
torchx_nightly-2024.7.3.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|