Fixed model weights path

BojarLab · Nov 6, 2024 · 92dbf59 · 92dbf59
1 parent c2b1e4d
commit 92dbf59
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CandyCrunch/prediction.py b/CandyCrunch/prediction.py
@@ -41,7 +41,7 @@
 if torch.cuda.is_available():
     device = "cuda:0"
 
-sdict = os.path.join(this_dir, 'sugarbase.pt')
+sdict = os.path.join(this_dir, 'candycrunch.pt')
 sdict = torch.load(sdict, map_location = device)
 sdict = {k.replace('module.', ''): v for k, v in sdict.items()}
 candycrunch = CandyCrunch_CNN(2048, num_classes = len(glycans)).to(device)