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')
Basic Models for PyTorch, with Unified Interface