search

chenyuntc / pytorch-book

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)
MIT License
11.99k stars 3.73k forks source link

第6章的models中resnet结构错误 #199

Open wyfqqq opened 4 years ago

wyfqqq commented 4 years ago

我发现第6章的models中resnet结构错误可能错误 self.layer1 = self.make_layer(64, 64, 3) self.layer2 = self.make_layer(64, 128, 4, stride=2) self.layer3 = self.make_layer(128, 256, 6, stride=2) self.layer4 = self.make_layer(256, 512, 3, stride=2) 我把它改为上述的

wzhings commented 4 years ago

我觉得你是对的！谢谢

YunhaoDu commented 3 years ago

我发现第6章的models中resnet结构错误可能错误 self.layer1 = self.make_layer(64, 64, 3) self.layer2 = self.make_layer(64, 128, 4, stride=2) self.layer3 = self.make_layer(128, 256, 6, stride=2) self.layer4 = self.make_layer(256, 512, 3, stride=2) 我把它改为上述的

确实有问题，谢谢。此外layer1的shortcut应该是identity操作，而不需要卷积