--- 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