I would like to change the feature extractor in the github code below
  

      github.com
  

  
    bbrattoli/JigsawPuzzlePytorch/blob/master/JigsawNetwork.py 2


      # -*- coding: utf-8 -*-
"""
Created on Wed Sep 13 15:57:01 2017

@author: Biagio Brattoli
"""
import torch
import torch.nn as nn
from torch import cat
import torch.nn.init as init

import sys
sys.path.append('Utils')
from Layers import LRN

class Network(nn.Module):

    def __init__(self, classes=1000):
        super(Network, self).__init__()





  This file has been truncated. show original

  

  
    
    
  

  


How to change self.conv to use something from the torchvision models directly ?
thank you