Skip to main content

# pytorch_modules

## Introduction

A neural network toolkit built on pytorch/opencv/numpy that includes neural network layers, modules, loss functions, optimizers, data loaders, data augmentation, etc.

## Features

  • Advanced neural network modules/loss functions/optimizers

  • Ultra-efficient trainer and dataloader that allows you to take full advantage of GPU

## Installation

sudo pip3 install pytorch_modules

or

sudo python3 setup.py install

## Usage

### pytorch_modules.utils

Includes a variety of utils for pytorch model training. See [woodsgao/pytorch_segmentation](https://github.com/woodsgao/pytorch_segmentation) as a tutorial.

### pytorch_modules.nn

This module contains a variety of neural network layers, modules and loss functions.

import torch from pytorch_modules.nn import ResBlock

# NCHW tensor inputs = torch.ones([8, 8, 224, 224]) block = ResBlock(8, 16) outputs = block(inputs)

### pytorch_modules.backbones

This module includes a series of modified backbone networks.

import torch from pytorch_modules.backbones import ResNet

# NCHW tensor inputs = torch.ones([8, 8, 224, 224]) model = ResNet(32) outputs = model.stages[0](inputs)

### pytorch_modules.datasets

This module includes a series of dataset classes integrated from pytorch_modules.datasets.BasicDataset which is integrated from torch.utils.data.Dataset . The loading method of pytorch_modules.datasets.BasicDataset is modified to cache data with LMDB to speed up data loading. This allows your gpu to be fully used for model training without spending a lot of time on data loading and data augmentation. Please see the corresponding repository for detailed usage.

Metadata

Release files for pytorch-modules 0.3.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for pytorch-modules 0.3.5
File Size Uploaded Provenance
pytorch_modules-0.3.5.tar.gz 14.9 kB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for pytorch-modules 0.3.5
File Interpreter ABI Platform
pytorch_modules-0.3.5-py2.py3-none-any.whl Python 3, Python 2 none any Details

Total release size: 53.5 kB

Release files / pytorch_modules-0.3.5.tar.gz

Download URL pytorch_modules-0.3.5.tar.gz
Size 14.9 kB
Tags Source
SHA-256 checksum
How to use checksums
5a2d3d5dcdea3d8021036e5fbb5dbe7cc2cbd36913c025b837e971724d6f2bc6
BLAKE2b-256 checksum
How to use checksums
4d809cbdfbb0978b522bf542876f57ca0281e21dfea53549c4f1865b8ca280b8
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/1.15.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.36.1 CPython/3.5.2

Release files / pytorch_modules-0.3.5-py2.py3-none-any.whl

Download URL pytorch_modules-0.3.5-py2.py3-none-any.whl
Size 38.5 kB
Tags Python 2 Python 3
SHA-256 checksum
How to use checksums
82648653e881a23c7a2790ab4a61d27fbd7c04fc1cd7800fe06e3892e4f22e35
BLAKE2b-256 checksum
How to use checksums
cac56bc94da943e1aa82b9fa38324b4ba59fb8ef2d71fd6b9512b62c147af3d0
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/1.15.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.36.1 CPython/3.5.2

Security for pytorch-modules 0.3.5

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page