I think it is correct as per this example

  
      
      My Journey with Deep Learning and Computer Vision – 30 Aug 19
  
  
    

Transfer Learning for Segmentation Using DeepLabv3 in PyTorch 120

In this post, I’ll be covering how to use a pre-trained semantic segmentation DeepLabv3 model for the task of road crack detection in PyTorch by using transfer learning. The same procedure ca…