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.

Files for torch-template, version 0.0.4
Filename, size File type Python version Upload date Hashes
Filename, size torch_template-0.0.4-py3-none-any.whl (49.1 kB) File type Wheel Python version py3 Upload date Hashes View hashes
Filename, size torch-template-0.0.4.tar.gz (36.9 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page