Thanks for the response @lewtun! Unfortunately, it did not work, and I received the same error with the same stack trace.
One thing I forgot to mention – the model seems to have run one training epoch successfully, and then crashed when doing the next step…
Not sure if that makes a difference, or adds any useful context.
Thanks again!