It works a little bit with a gain of 5% to use the TensorFlow version VGG16 in Pytorch (Attention: You have to change the names of variables, dimension order of tensors and normalization of data.) I didn’t figure it out. And I am going to turn to Tensorflow for better model optimization.