Skip to main content

Torch_Template - A PyTorch template with commonly used models and tools

Project description

torch_template

Documentation Status License

A python package for commonly used pytorch models, data loaders and utils.

Installation

For pip

pip install torch-template

For source

Clone the repo, cd into it and run pip install . command.

git clone https://github.com/misads/torch_template.git
cd torch_template
pip install .

For conda

source ~/anaconda3/bin/activate
conda activate <env>
python setup.py install

A configure file torch_template.egg-info will be generated in the repo directory. Copy torch_template and torch_template.egg-info to your site-packages folder.

Usage

Test if the package is successfully installed:

import torch_template as tt
from torch_template import torch_utils

Run:

#!-bash
tt-new

Enter your repo name, then a python project template will be created.

Documentation

The documentation webpage can be found here https://torch-template.readthedocs.io/en/latest/.

File structure

File structure
    .
    ├── network
    │     ├── base_model.py     :Define models, losses and parameter updating
    │     ├── norm.py           :Normalizations
    │     └── weights_init.py   :weights init
    
    ├── dataloader/             :Define Dataloaders
    ├── model_zoo               :Commonly used models
    └── utils
          └── torch_utils.py    :PyTorch utils

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

torch-template-0.0.4.tar.gz (36.9 kB view hashes)

Uploaded source

Built Distribution

torch_template-0.0.4-py3-none-any.whl (49.1 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page