@philokey, I will create a new branch for this so that the current pretrained won't be messed up. This issue was originally opened in https://github.com/ruotianluo/pytorch-resnet/issues/2