另外关于paddle代码改成pytorch代码的变换方式记录一下:
nn.Conv2D | nn.Conv2d |
---|---|
panddle.concat | torch.cat |
nn.BatchNorm2D | nn.BatchNorm2d |
nn.Layer | nn.Module |
bias_attr | bias |
将paddle代码转成pytorch代码时出现了这个问题。仔细查找代码,发现是这里的原因:
将其中的transpose改成permute即可啦!
另外关于paddle代码改成pytorch代码的变换方式记录一下:
nn.Conv2D | nn.Conv2d |
---|---|
panddle.concat | torch.cat |
nn.BatchNorm2D | nn.BatchNorm2d |
nn.Layer | nn.Module |
bias_attr | bias |
将paddle代码转成pytorch代码时出现了这个问题。仔细查找代码,发现是这里的原因:
将其中的transpose改成permute即可啦!