netloader 3.11.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.
@@ -0,0 +1,59 @@
1
+ Metadata-Version: 2.4
2
+ Name: netloader
3
+ Version: 3.11.0
4
+ Summary: Utility to generate and train PyTorch neural network objects from JSON files
5
+ Author-email: Ethan Tregidga <ethan.tregidga@epfl.ch>
6
+ License-Expression: MIT
7
+ Project-URL: Homepage, https://github.com/EthanTreg/PyTorch-Network-Loader
8
+ Project-URL: Issues, https://github.com/EthanTreg/PyTorch-Network-Loader/issues
9
+ Project-URL: ReadTheDocs, https://pytorch-network-loader.readthedocs.io/en/latest/
10
+ Classifier: Programming Language :: Python :: 3
11
+ Classifier: Operating System :: OS Independent
12
+ Requires-Python: >=3.12
13
+ Description-Content-Type: text/markdown
14
+ License-File: LICENSE.txt
15
+ Requires-Dist: numpy>=2.4.0
16
+ Requires-Dist: torch>=2.11.0
17
+ Requires-Dist: packaging>=24.0
18
+ Provides-Extra: flows
19
+ Requires-Dist: zuko; extra == "flows"
20
+ Provides-Extra: wandb
21
+ Requires-Dist: wandb; extra == "wandb"
22
+ Dynamic: license-file
23
+
24
+ # PyTorch Network Loader
25
+
26
+ PyTorch Network Loader is a utility that enables the easy creation of neural networks in PyTorch using JSON
27
+ configuration files.
28
+ The package automatically tracks layer output shapes, eliminating the need to manually calculate input dimensions for
29
+ each layer.
30
+ NetLoader also provides an architecture framework that allows for the easy architecture setup and training framework,
31
+ along with additional useful utilities to accelerate the use of neural networks in science.
32
+
33
+ See the [Wiki](https://pytorch-network-loader.readthedocs.io) for more information on how to use this package.
34
+ For a real-world example of this package, see [Bayesian-DARKSKIES](https://github.com/EthanTreg/Bayesian-DARKSKIES).
35
+
36
+ ## Requirements
37
+
38
+ ### Using Within Projects
39
+
40
+ - pip install `netloader @ git+https://github.com/EthanTreg/PyTorch-Network-Loader@LATEST-VERSION`[^1] to
41
+ `requirements.txt`
42
+ - Install using `pip install -r requirements.txt`
43
+ - Example of [InceptionV4](https://arxiv.org/abs/1602.07261) can be downloaded under
44
+ `./network_configs/inceptionv4.json` along with the composite layers in
45
+ `./network_configs/composite_layers/`
46
+
47
+ [^1]: To use normalising flows or weights & biases, `netloader` must be pip installed with the optional argument
48
+ `flows` and/or `wandb`:
49
+ `pip install netloader[flows] @ git+https://github.com/EthanTreg/PyTorch-Network-Loader@LATEST-VERSION`
50
+
51
+ ### Locally Running NetLoader
52
+
53
+ - Clone or download the repository
54
+ - Install dependencies:
55
+ `pip install -r requirements.txt`
56
+ - PyTorch's dependencies[^2]:
57
+ NVIDIA GPU with [CUDA Toolkit](https://developer.nvidia.com/cuda-toolkit-archive) ~= v12.1
58
+ [^2]: Only required for use with NVIDIA GPU, v11.8 is also supported, but requirements.txt will
59
+ try to install the v12.1 version
@@ -0,0 +1,36 @@
1
+ netloader/__init__.py,sha256=oxIVo2tzERGmbr-nhhVtrXElCYIHWIdWi2-n7rSza4s,1577
2
+ netloader/data.py,sha256=PZk1mki_seo4J9R-zYxTGp1BZMKntxQvJt28JOcuzv0,35723
3
+ netloader/loss_funcs.py,sha256=vcMZOYn_YSPoUfQNuRsQnicXW9zEPVnEyHVHG-i8iIw,6487
4
+ netloader/network.py,sha256=D-OUzVXEZN4mRTc7Xs_M0lR1hrmEZhmIDc6ynrqFpQI,31043
5
+ netloader/schedulers.py,sha256=p6E-BnbYmEc9032xdKWYFRrZP48MIyXDj0DwRSYvof0,11653
6
+ netloader/transforms.py,sha256=lm_edz4qPyp0D3sKFmnQ9muRjWr5tgIuWEfJKy4KNl4,28945
7
+ netloader/architectures/__init__.py,sha256=Xb-kC0w80HRSFHN76pnURalyGhaSM5ndCQhEAMCIX28,1466
8
+ netloader/architectures/base.py,sha256=kMu7N2AfvfJ4oUNrZlaRmK-dZrq-XUr-dSQnB-9EQ50,50108
9
+ netloader/architectures/encoder_decoder.py,sha256=OfZ2jvFNZBgkN8PHxOVFKXZcNSbm9G2gVLw-MYG9X-8,24351
10
+ netloader/architectures/flows.py,sha256=OWZTeU6Q9TieUgOMeZWbTWbXc3YWWABKNdYlhdKvv4c,19613
11
+ netloader/architectures/utils.py,sha256=AUJhg58Xc1si9BJEd2MAlX_rr9QQukjXvCAxS13yq2c,5682
12
+ netloader/layers/__init__.py,sha256=qGYt7C2GN_wbk_9xkjuGfs5-aYPnPrkb6NZD8out7ds,2071
13
+ netloader/layers/base.py,sha256=Uk3RYUbWrD3C_EJIbGaBfKlGlpsBwJHxtTYUwDMhU0Q,12089
14
+ netloader/layers/blocks.py,sha256=CvN_uRuLPDVmOMMtTtozavW3sPbSux9di1inxp_Na_0,4653
15
+ netloader/layers/convolutional.py,sha256=slJgTWfHeXSb58pmsvaqAt4r7KntMMBdNbvj1wk5TtQ,37806
16
+ netloader/layers/flows.py,sha256=eexylw9_JCxGOy65C6r5eHzT5mIX4a0Ns3ArfGmbOqU,3933
17
+ netloader/layers/linear.py,sha256=VMIiUu-GHbN20N8eUbbTsiucy-PVQh48dF3mwbvDsWc,16008
18
+ netloader/layers/misc.py,sha256=DcnQnD5p2GaVTPNHvedxX8LN9V2oMVNxFsjqPFXkeTE,28774
19
+ netloader/layers/multi_layer.py,sha256=ofxtyNN7h8zicZCbF9Dgv6DTOh8Vjki_Y8kRk2wXviA,15657
20
+ netloader/layers/pooling.py,sha256=NvDYEnH_5-2C5A0FrcXrg8maFpWv_8DGjvngVlmFvsY,10217
21
+ netloader/layers/recurrent.py,sha256=YvCaV9ciWnmsBtWSrFLdwWtTFBXUSPGqWLUUhwq-0wo,5778
22
+ netloader/layers/utils.py,sha256=q-7Y9Yffi-yjr_Tw_qX8eJ0qUBAFDJVQcjPXOmBrS-U,4883
23
+ netloader/models/__init__.py,sha256=bkAWRORq37R5yu7eZNmQOJd448VxsdoW7kCK5TVOZfI,424
24
+ netloader/models/convnext.py,sha256=5bIl47d72EKCNE9odTfEcceyVhO_R-aSLIQTIdtLiNg,19687
25
+ netloader/models/misc.py,sha256=eZC_gAKgaF-EPXQdl_2gvw1OBcfuWTX5qOuRFVFzk18,8353
26
+ netloader/networks/__init__.py,sha256=5YYbIX1lT67uZppdfYMCO9bEFA8tych2jAaq7_Z8bWc,1150
27
+ netloader/utils/__init__.py,sha256=EobamjFYKEJtmRHcbK4UNTj7AvkeEkfLxeH1kUKnF0Y,975
28
+ netloader/utils/configs.py,sha256=_F9-QWRe6AVSqqj3m3L2Vc2G0pffObBG3stUjHFU_9k,7363
29
+ netloader/utils/transforms.py,sha256=BMv2d54AXCe3AccHIHtcysZzD8cUN-ofNpPZSpydZKQ,466
30
+ netloader/utils/types.py,sha256=90Vk7vypoDerJxMcE7JT7_H2vi8Lg67Cup5OpKjsQkU,3778
31
+ netloader/utils/utils.py,sha256=eyQ0JgR5nh9ZSzLv3id6LprfP4eKO5EgnYmSxs3RgIY,22610
32
+ netloader-3.11.0.dist-info/licenses/LICENSE.txt,sha256=FKqPqfbsw4ai8nKZTeP111QfClT0ijPawH90VOA_T3M,1092
33
+ netloader-3.11.0.dist-info/METADATA,sha256=Y3b9rCrUv4w-VMxPdm0USyEMN8yR_OGsWBnbJGp32zA,2708
34
+ netloader-3.11.0.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
35
+ netloader-3.11.0.dist-info/top_level.txt,sha256=GgpdZcMKPdbfX4i54fkjVQTgY99r8AyH8lqkTVuAB74,10
36
+ netloader-3.11.0.dist-info/RECORD,,
@@ -0,0 +1,5 @@
1
+ Wheel-Version: 1.0
2
+ Generator: setuptools (83.0.0)
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
5
+
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2023 Ethan Tregidga
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -0,0 +1 @@
1
+ netloader