From e145104024638298ff8926e720619ca427b3b3c6 Mon Sep 17 00:00:00 2001 From: GardevoirX Date: Fri, 31 May 2024 01:27:18 +0200 Subject: [PATCH] Updata inference script --- inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/inference.py b/inference.py index 08eb935..54dd3ac 100644 --- 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)