yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers
MIT License
29.83k stars 8.09k forks source link

[Bug fix] redundant layers in ResNet #141

Closed keineahnung2345 closed 5 years ago

keineahnung2345 commented 5 years ago

In https://github.com/yunjey/pytorch-tutorial/blob/master/tutorials/02-intermediate/deep_residual_network/main.py#L115, it defined a length 4 layers. But in https://github.com/yunjey/pytorch-tutorial/blob/master/tutorials/02-intermediate/deep_residual_network/main.py#L84, it only uses layers[0] and layers[1]. So the last entry of [2,2,2,2] should be redundant.

yunjey commented 5 years ago

@keineahnung2345 You are right. Thanks very much.