Imported from https://github.com/EdenMelaku/Transfer-Learning-Pytorch-Implmentation