def init_model(path, num_classes, num_gpus):
model = models.vgg16_bn(pretrained=False)
num_ftrs = model.classifier[6].in_features
model.classifier[6] = nn.Linear(num_ftrs, num_classes)
model = MMDataParallel(model, device_ids=range(num_gpus)).cuda()
model.load_state_dict(torch.load(path))
return model
