final submission

README.md

@@ -7,29 +7,37 @@ This repository contains re-Implementation of https://arxiv.org/abs/2103.06255
 
 
 The authors of _Involution: Inverting the Inherence of Convolution for Visual Recognition_ propose a novel involutional layers, which aims to enhance the representation power of convolutional neural networks by inverting the inherent properties of convolution operations. As such these kernals are channel agnostic and spatial specific.
+
 ### Folders
 
 ```models``` folder contains the main backbone implementations of models used as well as classification heads and lightning class for easy training and logging
 
+```slides``` contains presentation slides
+
+```data``` contains the data module and custom dataset
+
 ### Training
 
 ```
 git clone https://github.com/thatblueboy/involution.git #clone the repo
 git checkout submission_branch #change to the submission branch
 ```
-Edit the ```train.py``` file in the main folder. Here you can change various Hyperparameters in the config dict. Note that changing lr_scheduler will require corresponding chnage in lr_sceduler_kwargs.
-Note: We use a random split split on Caltech256. For uniformity we store this split in the data_module.pth and load it for every training run.
-This behaviour could be changed by setting the 'data_module_path' value in the config dict = None.
+Edit the ```train.py``` file in the main folder. 
+
+- Here you can change various Hyperparameters in the config dict. Note that changing ```lr_scheduler``` will require corresponding chnage in ```lr_sceduler_kwargs```.
+
+- Note: We use a random split split on Caltech256. For uniformity we store this split in the data_module.pth and load it for every training run. This behaviour could be changed by setting the ```'data_module_path'``` value in the config dict to ```None```.
+
 
-To switch from training to testing mode, change the last line in the train.py from
+- To switch from training to testing mode, change the last line in the train.py from
 ```
 trainer.fit(model, data_module)
 ```
 to
 ```
 trainer.test(test, data_module)
 ```
-after making all the necessary changes do:
+- After making all the necessary changes do:
 ```
 wandb login
 python train.py