Since this appears to be a classification task, the MNIST or ImageNet examples:
examples/main.py at master · pytorch/examples (github.com) 1
examples/main.py at master · pytorch/examples (github.com)
should be a good starting point.