a419775258

可以提供一个最小复现吗？或者提供一下从哪个 op 开始，梯度的值就和 torch 对不上了。

我理解，梯度应该是从网络中的某一个算子开始，和 torch 计算的结果开始出现比较大的差距，这个计算结果的差距和网络应该是无关的。现在需要您提供一下具体是哪个算子，我们才好修复。 如果您发现 SynthesisLayer 和 ToRGBLayer 单独都没有问题，但是两者一组合就和 torch 不同，也请您提供一个最小复现。不是在 SynthesisNetwork 这个网络中，而是两者简单的组合。

也可以 `export CUDA_VISIBLE_DEVICES=''` 使用 CPU 计算。