Oh thanks both, will do that.
But is there not any code like this one for pytorch,

  
      github.com
  
  
    keras-team/keras-applications/blob/master/keras_applications/vgg16.py 100
"""VGG16 model for Keras.

# Reference

- [Very Deep Convolutional Networks for Large-Scale Image Recognition](
    https://arxiv.org/abs/1409.1556)

"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import os

from . import get_submodules_from_kwargs
from . import imagenet_utils
from .imagenet_utils import decode_predictions
from .imagenet_utils import _obtain_input_shape

preprocess_input = imagenet_utils.preprocess_input


  This file has been truncated. show original

  
  
    
    
  
  


It will reduce work, I am too lazy lol.