Update README.md

README.md

@@ -12,6 +12,10 @@ license: mit
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
+Training details:
+1. This model is trained on CIFAR10 with following classes: ['plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
+2. It is trained on 24 epochs using One Cycle Policy using PyTorch Lightning
+
 Features of spaces app:
 1. User can decide on whether he/she wants to see GradCAM images and how many, and from which layer, allow opacity change as well
 2. User can ask whether he/she wants to view misclassified images, and how many (maximum limit is 50)