Create a sample project using pytorich lighting

Job ID: 33570244

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

TCreate 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