torch.load を map_location=torch.device('cpu') とともに使用して、ストレージを CPU にマッピングしてください。
NoSuchKey
おすすめ
転載: blog.csdn.net/qq_39691492/article/details/121120909
おすすめ
ランキング