In case you missed it, here's a link to the full training example that you can run yourself: https://www.paperspace.com/console/jobs/jvqssfqawv5zn/logs

Inference example: https://www.paperspace.com/console/jobs/js4mqzm91fj2lg

Disclosure: I work on Paperspace