Basic template for Pytorch based DL training.
model.py : Declare the network and various layers in here.
config.py : Configure the training and testing parameters.
main.py : Contains train, test and main functions.
Run main.py to train and test the model on MNIST dataset.