Create a sample project using pytorich lighting on GPU environment

Job ID: 33577102

Budget: ₹1,000 – ₹3,500 INR

Create a code sample that uses PyTorch Lightning and sstack ( 3rd party library ) to train a simple model

Requirements:

1. It’s allowed to use one of the existing examples provided by PyTorch Lightning from their documentation or repository (e.g. MNIST). You just need to adapt it to run with sstack.
2. The sample should allow running the training workflow without GPU, with a single GPU, and with multiple GPU. For example, you can use a sstack variable to control it.
3. Be sure to check that the training actually works in all cases (with a single GPU, with multiple GPUs, and with no GPUs).

Sstack is basically allows you to automate training workflows, version and reuse data and models using a cloud vendor of your choice. We will be using AWS cloud service for this