阿刚的代码进阶之旅

摘要： Pytorch中的split问题： 1、使用torch.nn.Conv2d中有个参数是groups会将输入的feature map分组，此处需要注意的一点是分组之后各组的feature map的channel数目是相同的。 2、另外一种方式用torch.split函数将输入的feature map先 阅读全文

摘要： 对于Pytorch的结果复现，在不同版本的pytorch或者不同的硬件平台之间，不能够保证结果是完全一致的，即使在CPU和GPU上采用相同的随机数种子。 关于torch.backend.cudnn.benchmark， （CuDNN backend是什么） 是相对来说确定性的模式。 包括Conv2d 阅读全文