search

dailenson / SDT

This repository is the official implementation of Disentangling Writer and Character Styles for Handwriting Generation (CVPR23).
MIT License
956 stars 81 forks source link

Mac M1 不支持Pytorch的CUDA. 因为木有英伟达显卡 #40

Open MaySoon2020 opened 8 months ago

TxcA commented 8 months ago

找到全部.cuda() 修改为 to('cpu')

dailenson commented 8 months ago

在M1芯片的GPU上训练，试试将源代码中的torch.device("cuda")替换成torch.device("mps")？