Update README.md

README.md

@@ -7,12 +7,9 @@ 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.
-### Setup
-```
-pip install torch torchvision
-pip install wandb
-pip install lightning
-```
+### Folders
+
+```models ``` folder contains the main backbone implementations of models used as well as classification heads and lightning class for easy training and logging
 
 ### Training