Skip to content

Latest commit

 

History

History
23 lines (17 loc) · 769 Bytes

File metadata and controls

23 lines (17 loc) · 769 Bytes

SpectralLayersPyTorch

Trainable linear spectral layers for PyTorch

Implements trainable spectral layers for PyTorch that can be initialized as 1-D & 2-D DCT and DFT transformations as shown in paper.

Install

The package can be installed as follows:

pip install git+https://github.com/NarayanSchuetz/SpectralLayersPyTorch.git

Attribution

@misc{alberti2019trainable,
      title={Trainable Spectrally Initializable Matrix Transformations in Convolutional Neural Networks}, 
      author={Michele Alberti and Angela Botros and Narayan Schuez and Rolf Ingold and Marcus Liwicki and Mathias Seuret},
      year={2019},
      eprint={1911.05045},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
   }