Name 'vgg' is not defined

Hi guys,

I just found an error - name ‘vgg’ is not defined - when I ran the training codes. Listed are part codes in the training file. Any ideas are welcome. Cheers!

import os
import torch
import numpy as np
import scipy
import sys
from import *

VGG is a architecture-defined python file in the Model folder

from Model.VGG import *

import torch.nn as nn
import torch.optim as optim
import torch.nn.functional as F
import torch.autograd as autograd
from import Dataset, DataLoader
#from PIL import Image
import matplotlib.pyplot as plt
from torchvision import datasets, models, transforms

class Train:
def init(self, number_classes = 2):
Init Dataset, Model and others
root_path = “/home/*****”
path_prefix = “VGG16”
self.save_prefix = path_prefix

    self.num_classes = number_classes
    #self.pretrained = None
    self.aff_dataset = ImageData(root_path=root_path, train_mode = "train")
    dataloader = DataLoader(self.aff_dataset, batch_size=batch_size, shuffle=True, num_workers=8)

    # create a Net
    vgg = VGG()
    vgg16 = models.vgg16(pretrained=True)
    pretrained_dict = vgg16.state_dict()
    model_dict = vgg.state_dict()
    pretrained_dict =  {k: v for k, v in pretrained_dict.items() if k in model_dict}