import vgg
class VGG(nn.Module):
'''
VGG model
'''
def __init__(self, features):
super(VGG, self).__init__()
self.features = features
self.classifier = nn.Sequential(
nn.Dropout(),
nn.Linear(512, 512),
nn.ReLU(True),
nn.Dropout(),
nn.Linear(512, 512),
nn.ReLU(True),
nn.Linear(512, 10),
)
# Initialize weights
def forward(self, x):
x = self.features(x)
x = x.view(x.size(0), -1)
return x
ModuleNotFoundError: No module named ‘vgg’
help me plzz