metadata
license: mit
library_name: pytorch
pipeline_tag: image-classification
tags:
- pytorch
Uses PyTorch and a combination of layers to create a fairly simple network. You can check model.py for the layers and training code