torch-template 0.0.3

Torch_Template - A PyTorch template with commonly used models and tools

torch_template

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

Installation

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

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

For pip

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

import torch_template
from torch_template import misc_utils as utils
from torch_template import torch_utils

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
          ├── misc_utils.py     :System utils
          └── torch_utils.py    :PyTorch utils