AssertionError: Torch not compiled with CUDA enabled. 出錯程式碼:. index = torch.randperm(batch_size).cuda(). 修改為如下即可:
確定! 回上一頁