Training a Simple PyTorch Model with Custom Dataset

Prompt

Result