DeepLearning deep learning using pytorch different initialization batch normalization transfer learning on flower dataset