Say suppose you have a class where your architecture is defined like:
change
model = resnet18(pretrained=True, num_classes=10)

to
model = Your_Net(num_classes=10)

In your code I guess
model = ModifiedSqueezeNetModel(num_classes = .... )