diudiu

@Hussainflr If there is no error message but just gets stuck, I guess it's because the memory is too full to keep on? You can check the memory usage when...

@Hussainflr I also used Pytorch 0.4.1 and python 3.7

I guess it is because that the paper result is pre-trained on ImageNet?