ValueError loading dataset in SageMaker notebook

pytorch_version='1.6',

This sample seems to assume torch 1.6. Either way, it’s a very old version…
It’s 2.5.1 now…
Even for stability, 2.2 is more common.
2.3 is a minefield.