torch-basic-models

Basic Models for PyTorch, with Unified Interface

Installation

Need Python 3.6+.

pip install torch-basic-models

Usage

import torch_basic_models

ResNet = torch_basic_models.ResNet

# or

import box

ResNet = box.load(name='ResNet', tag='model')