my fork of prototorch_models

Go to file

Jensun Ravichandran 5a1ef841d3 Update mnist example		2021-04-21 16:28:20 +02:00
examples	Update mnist example	2021-04-21 16:28:20 +02:00
prototorch/models	Add glvq model	2021-04-21 14:51:34 +02:00
.gitignore	Update gitignore	2021-04-21 14:54:22 +02:00
LICENSE	Initial commit	2021-04-21 13:13:28 +02:00
README.md	Update readme	2021-04-21 15:02:55 +02:00
setup.py	Update installation dependencies	2021-04-21 14:51:08 +02:00

ProtoTorch Models

Pre-packaged prototype-based machine learning models using ProtoTorch and PyTorch-Lightning.

Installation

To install this plugin, simple install ProtoTorch first by following the installation instructions there and then install this plugin by doing:

git clone https://github.com/si-cim/prototorch_models.git && cd prototorch_models
pip install -e .

The plugin should then be available for use in your Python environment as prototorch.models.