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.