Train the model by running: 
cmd :: 'python train.py'
The model is stored in 'snapshot.pth' file.
We can now test individual images(test.jpg in this example) using test.py script by running:
cmd:: 'python test.py test.jpg'