minor changes

train_tongue_diagnosis.py

@@ -77,11 +77,11 @@ def get_model(model_name):
     elif model_name == 'vgg19_bn':
         model = models.vgg19_bn(pretrained=True)
         model.classifier[6] = torch.nn.Linear(in_features=4096, out_features=NUM_CLASSES, bias=True)
-        return models.to(device)
+        return model.to(device)
     elif model_name == 'inception_v3':
         model = models.inception_v3(pretrained=True)
         model.fc = torch.nn.Linear(in_features=2048, out_features=1000, bias=True)
-        return models.to(device)
+        return model.to(device)
     else:
         raise ValueError('Unsupport model: {0}', model_name)