How to change the code to use feature extractor from torchvision

ramdhan1989 · February 2, 2022, 12:57am

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

bbrattoli/JigsawPuzzlePytorch/blob/master/JigsawNetwork.py

# -*- 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