my fork of prototorch_models

Go to file

Jensun Ravichandran fe36e5fad9 Add partial metric/hparam features [BROKEN STATE]		2021-04-21 19:16:57 +02:00
examples	Add partial metric/hparam features [BROKEN STATE]	2021-04-21 19:16:57 +02:00
prototorch/models	Add partial metric/hparam features [BROKEN STATE]	2021-04-21 19:16:57 +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	Add partial metric/hparam features [BROKEN STATE]	2021-04-21 19:16:57 +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.