Install prerequisites with:
pip install -r requirements.txt
Make sure to comment out "e2cnn/nn/modules/r2_conv/r2convolution.py" Line number 176 and 177.
The main script offers many options; here are the most important ones:
usage: python main.py --model_name {model name} --dataset {dataset: CIFAR10, CIFAR100, rot-MNIST} --num_classes {num_classes} --epochs {number of epochs to train}
Please cite our paper in your publications if it helps your research. Even if it does not,and you want to make us happy, do cite it :)
@INPROCEEDINGS{9506158,
author={Chavan, Arnav and Bamba, Udbhav and Tiwari, Rishabh and Gupta, Deepak},
booktitle={2021 IEEE International Conference on Image Processing (ICIP)},
title={Rescaling CNN Through Learnable Repetition of Network Parameters},
year={2021},
volume={},
number={},
pages={754-758},
doi={10.1109/ICIP42928.2021.9506158}}
This project is licensed under the MIT License.