Open gallexy opened 13 hours ago
感谢开源这个项目，获益良多！ 跑了几遍这个程序，有一点比较好奇：好像训练中大多计算没有利用GPU加速， 反而有些代码强制使用cpu运算，如下： with to_cpu(*self.sample_modules) 不知是什么缘故。是专门这样设计的吗？
感谢开源这个项目，获益良多！ 跑了几遍这个程序，有一点比较好奇：好像训练中大多计算没有利用GPU加速， 反而有些代码强制使用cpu运算，如下： with to_cpu(*self.sample_modules) 不知是什么缘故。是专门这样设计的吗？