update

contrast/config.py.bak

@@ -0,0 +1,21 @@
+import torch
+import torchvision.transforms as T
+
+class Config:
+    host = "192.168.1.28"
+    port = "19530"
+
+    embedding_size = 256
+    img_size = 224
+    test_transform = T.Compose([
+        T.ToTensor(),
+        T.Resize((img_size, img_size)),
+        T.ConvertImageDtype(torch.float32),
+        T.Normalize(mean=[0.5], std=[0.5]),
+    ])
+
+    # test_model = "checkpoints/resnet18_our388.pth"
+    test_model = "checkpoints/mobilenetv3Large_our388_noPara.pth"
+    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+config = Config()