netloader 3.11.0__tar.gz

netloader-3.11.0/LICENSE.txt

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 Ethan Tregidga
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

netloader-3.11.0/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.4
+Name: netloader
+Version: 3.11.0
+Summary: Utility to generate and train PyTorch neural network objects from JSON files
+Author-email: Ethan Tregidga <ethan.tregidga@epfl.ch>
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/EthanTreg/PyTorch-Network-Loader
+Project-URL: Issues, https://github.com/EthanTreg/PyTorch-Network-Loader/issues
+Project-URL: ReadTheDocs, https://pytorch-network-loader.readthedocs.io/en/latest/
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE.txt
+Requires-Dist: numpy>=2.4.0
+Requires-Dist: torch>=2.11.0
+Requires-Dist: packaging>=24.0
+Provides-Extra: flows
+Requires-Dist: zuko; extra == "flows"
+Provides-Extra: wandb
+Requires-Dist: wandb; extra == "wandb"
+Dynamic: license-file
+
+# PyTorch Network Loader
+
+PyTorch Network Loader is a utility that enables the easy creation of neural networks in PyTorch using JSON
+configuration files.
+The package automatically tracks layer output shapes, eliminating the need to manually calculate input dimensions for
+each layer.
+NetLoader also provides an architecture framework that allows for the easy architecture setup and training framework,
+along with additional useful utilities to accelerate the use of neural networks in science.
+
+See the [Wiki](https://pytorch-network-loader.readthedocs.io) for more information on how to use this package.  
+For a real-world example of this package, see [Bayesian-DARKSKIES](https://github.com/EthanTreg/Bayesian-DARKSKIES).
+
+## Requirements
+
+### Using Within Projects
+
+- pip install `netloader @ git+https://github.com/EthanTreg/PyTorch-Network-Loader@LATEST-VERSION`[^1] to
+  `requirements.txt`
+- Install using `pip install -r requirements.txt`
+- Example of [InceptionV4](https://arxiv.org/abs/1602.07261) can be downloaded under
+  `./network_configs/inceptionv4.json` along with the composite layers in
+  `./network_configs/composite_layers/`
+
+[^1]: To use normalising flows or weights & biases, `netloader` must be pip installed with the optional argument
+`flows` and/or `wandb`:
+`pip install netloader[flows] @ git+https://github.com/EthanTreg/PyTorch-Network-Loader@LATEST-VERSION`  
+
+### Locally Running NetLoader
+
+- Clone or download the repository
+- Install dependencies:
+  `pip install -r requirements.txt`
+- PyTorch's dependencies[^2]:  
+  NVIDIA GPU with [CUDA Toolkit](https://developer.nvidia.com/cuda-toolkit-archive) ~= v12.1
+  [^2]: Only required for use with NVIDIA GPU, v11.8 is also supported, but requirements.txt will
+  try to install the v12.1 version

netloader-3.11.0/README.md

@@ -0,0 +1,36 @@
+# PyTorch Network Loader
+
+PyTorch Network Loader is a utility that enables the easy creation of neural networks in PyTorch using JSON
+configuration files.
+The package automatically tracks layer output shapes, eliminating the need to manually calculate input dimensions for
+each layer.
+NetLoader also provides an architecture framework that allows for the easy architecture setup and training framework,
+along with additional useful utilities to accelerate the use of neural networks in science.
+
+See the [Wiki](https://pytorch-network-loader.readthedocs.io) for more information on how to use this package.  
+For a real-world example of this package, see [Bayesian-DARKSKIES](https://github.com/EthanTreg/Bayesian-DARKSKIES).
+
+## Requirements
+
+### Using Within Projects
+
+- pip install `netloader @ git+https://github.com/EthanTreg/PyTorch-Network-Loader@LATEST-VERSION`[^1] to
+  `requirements.txt`
+- Install using `pip install -r requirements.txt`
+- Example of [InceptionV4](https://arxiv.org/abs/1602.07261) can be downloaded under
+  `./network_configs/inceptionv4.json` along with the composite layers in
+  `./network_configs/composite_layers/`
+
+[^1]: To use normalising flows or weights & biases, `netloader` must be pip installed with the optional argument
+`flows` and/or `wandb`:
+`pip install netloader[flows] @ git+https://github.com/EthanTreg/PyTorch-Network-Loader@LATEST-VERSION`  
+
+### Locally Running NetLoader
+
+- Clone or download the repository
+- Install dependencies:
+  `pip install -r requirements.txt`
+- PyTorch's dependencies[^2]:  
+  NVIDIA GPU with [CUDA Toolkit](https://developer.nvidia.com/cuda-toolkit-archive) ~= v12.1
+  [^2]: Only required for use with NVIDIA GPU, v11.8 is also supported, but requirements.txt will
+  try to install the v12.1 version

netloader-3.11.0/netloader/__init__.py

@@ -0,0 +1,61 @@
+"""
+Package information, creates the logger and adds netloader classes to PyTorch safe globals
+"""
+import sys
+import logging
+import warnings
+
+
+__version__ = '3.11.0'
+__author__ = 'Ethan Tregidga'
+logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.WARNING)
+warnings.filterwarnings('once', category=DeprecationWarning, module=r'^netloader(\.|$)')
+warnings.filterwarnings(
+    'once',
+    category=PendingDeprecationWarning,
+    module=r'^netloader(\.|$)',
+)
+
+
+if sys.version_info < (3, 12):
+    warnings.warn(
+        f'Python version ({sys.version_info.major}.{sys.version_info.minor}) is '
+        f'deprecated, please upgrade to Python 3.12 or higher',
+        DeprecationWarning,
+    )
+
+try:
+    import torch
+
+    from netloader.utils import safe_globals
+    from netloader import (
+        layers,
+        models,
+        architectures,
+        utils,
+        data,
+        loss_funcs,
+        network,
+        schedulers,
+        transforms,
+    )
+
+
+    # Adds PyTorch Network Loader classes to list of safe PyTorch classes when loading saved
+    # architectures
+    safe_globals(__name__, [models, architectures, loss_funcs, schedulers, transforms])
+    torch.serialization.add_safe_globals([network.Network, network.CompatibleNetwork])
+
+    __all__ = [
+        'utils',
+        'layers',
+        'models',
+        'architectures',
+        'data',
+        'network',
+        'loss_funcs',
+        'schedulers',
+        'transforms',
+    ]
+except (ModuleNotFoundError, ImportError):
+    pass

netloader-3.11.0/netloader/architectures/__init__.py

@@ -0,0 +1,48 @@
+"""
+Collects all architectures
+"""
+from typing import Any
+from warnings import warn
+
+from netloader.architectures.utils import UtilityMixin
+from netloader.architectures.base import BaseArchitecture, load_arch
+from netloader.architectures.encoder_decoder import Autoencoder, Decoder, Encoder
+
+
+__all__ = [
+    'BaseArchitecture',
+    'UtilityMixin',
+    'Autoencoder',
+    'Decoder',
+    'Encoder',
+    'load_arch',
+]
+
+_optional_imports: dict[str, set[str]] = {'flow': {'NormFlow', 'NormFlowEncoder'}}
+
+try:
+    from netloader.architectures.flows import NormFlow, NormFlowEncoder
+    __all__.extend(_optional_imports['flow'])
+except (ModuleNotFoundError, ImportError) as e:
+    pass
+
+
+def __getattr__(name: str) -> Any:
+    if name == 'BaseNetwork':
+        warn(
+            'BaseNetwork is deprecated, please use BaseArchitecture instead',
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        return BaseArchitecture
+    if name == 'load_net':
+        warn(
+            'load_net is deprecated, please use load_arch instead',
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        return load_arch
+    if name not in __all__ and name in _optional_imports['flow']:
+        raise ImportError(f'Cannot import {name}, normalising flow architectures require the '
+                          f'package Zuko')
+    raise AttributeError(f"module {__name__} has no attribute {name}")