PyTorchModelFromScratch training script

[hnky]

Describe the bug

After submitting the training job:
run = experiment.submit(config)

The training script gives an error:
User program failed with HTTPError: HTTP Error 404: Not Found

Driver log attached:
70_driver_log.txt

[MaxStrange]

Looks like this line:

Downloading: "https://github.com/pytorch/vision/archive/master.zip" to /root/.cache/torch/hub/master.zip

Fails with a 404. Looks like they changed from "master" to "main". So:

"https://github.com/pytorch/vision/archive/main.zip"

Should work. I'll look at where this line is happening in the script and see what I can do about updating it. Thanks for pointing this out.