Update README.md

README.md

@@ -17,3 +17,21 @@ 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}
 ```
+
+## Citation
+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}}
+
+
+## License
+
+This project is licensed under the MIT License.