EinsumNetwork

Getting started

  • Install dependencies in the requirements.txt

Usage

  • run the demos including demo_mnist, demo_2dmnist, demo_cifar, you can see the trained models in the directory "models" and sample images in "samples"
  • see the frequency domain analysis in the directory "cifar_fft_plots"