Imported from https://github.com/pravitc/Neural-style-transfer-using-Pytorch