Closed MeaninglessAI closed 5 years ago
在ResNet.py文件中,有如下注释:
我阅读过代码后,觉得这个传入的x维度应该是[batch_size,seq_len,channels,height,width] 这样的话,下面这句代码才合理: x = x.view(bt, x.size(2), x.size(3), x.size(4)) 否则,没有理由将x resize成bt,batch_size*channels,很莫名其妙
@yangjinray 你是对的,我当时理解有问题。。
感谢解答
@yangjinray 也感谢你指出错误 :smile:
在ResNet.py文件中,有如下注释:
我阅读过代码后,觉得这个传入的x维度应该是[batch_size,seq_len,channels,height,width] 这样的话,下面这句代码才合理: x = x.view(bt, x.size(2), x.size(3), x.size(4)) 否则,没有理由将x resize成bt,batch_size*channels,很莫名其妙