adding a training loop

[SantoshParit]

I am planning following subtasks:

1. Create a loop to trrain the ReseNet -- For now, trying to run on CIFAR10 - Later we can try with different datasets. -- Once the training and validation code starts working, I want to add something like confusion matrix.

@deepukr007 @kbpranav @Sohambasu07 @sushil579