New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

CtrlC中断训练后，重新执行训练命令后报错 #32

Open

wormseu opened this issue May 6, 2023 · 0 comments

wormseu commented May 6, 2023

环境是mac os M1 pro，使用mps进行训练，训练已经顺利生成了超过10个以上的checkpoint，但中断后再重新执行训练命令后，代码中出现以下报错：

 File "/Users/username/opt/anaconda3/envs/OCR_trainer/lib/python3.8/site-packages/torch/serialization.py", line 267, in default_restore_location
    raise RuntimeError("don't know how to restore data location of "
RuntimeError: don't know how to restore data location of torch.storage.UntypedStorage (tagged with mps:0)

请问如何解决

The text was updated successfully, but these errors were encountered:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment