前往
大廳
主題

AI Train Bug 可能解法 indices should be either on cpu or on the same device as the indexed tensor (cpu)

歐雷 | 2022-12-10 19:01:52 | 巴幣 0 | 人氣 894

AI Train 有時候會出現 indices should be either on cpu or on the same device as the indexed tensor (cpu) ,這裡提供一個可能的解法。

ddpm.py
第 1030 列上面新增
self.logvar = self.logvar.to(self.device)


創作回應

追蹤 創作集

作者相關創作

更多創作