torch_template

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