Updata inference script

GardevoirX · May 30, 2024 · e145104 · e145104
1 parent ae077bb
commit e145104
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inference.py b/inference.py
@@ -19,8 +19,8 @@ def parse_args():
 if __name__ == "__main__":
     args = parse_args()
 
-    model = torch.load("model.pt")
-    parameters = torch.load("auxiliary.pt")
+    model = torch.load("models/model.pt")
+    parameters = torch.load("models/auxiliary.pt")
     model.eval()
     test = CNSDataset(
         args.data_file, transform=DescriptorGenerator(AVAILABLE_DESCRIPTORS)