Original Setup has been taken from here, arxiv, and slight modifications (incl. FNO/CNO implementation) of the codebase have been performed. For CNO implementation refer to this repo, ViTs have been fine-tuned using code from this repo.
To reproduce experiments for everything other than ViT, simply run the (commented-out) parts of the run script.
To reproduce ViT finetuning, clone this repo, first run preprocess_FMNIST.py, then the run script, and for evalutation, run vit_eval_script.py.