ankitshah009 / phinet_pl

PhiNet module for PyTorch and PyTorch Lightning.

---

This repo contains the code for PhiNets, a scalable backbone for low-power AI at the edge

Description

This is a first version of the public repo. It will be reorganized with specific instructions to reproduce the results of the paper above (with data download, etc). In the meantime, if you are willing to use the PhiNets in your signal processing pipelines, you can import them in PyTorch from this repo (soon pip!) or use them in keras from this gist.

Example: cifar10

To train PhiNets for cifar10, you can run:

python __main__.py cifar10 data

Citation

@article{phinet,
Author = {Francesco Paissan and Alberto Ancilotto and Elisabetta Farella},
Title = {{PhiNets: a scalable backbone for low-power AI at the edge}},
Year = {2021},
Eprint = {arXiv:2110.00337},
}