From f6a636e6f67df6be84c3f0444616595ee0fb44f5 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 30 Sep 2024 11:04:35 +0200
Subject: [PATCH] fix save preds with file
---
main_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index 0cfc1cc..994b601 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -284,7 +284,7 @@ def save_pred(model, data_val, forward, output_path, file = False):
for seq, charge, rt, intensity, files in data_val:
rt, intensity = rt.float(), intensity.float()
if torch.cuda.is_available():
- seq, charge, rt, intensity, file = seq.cuda(), charge.cuda(), rt.cuda(), intensity.cuda(), file.cuda()
+ seq, charge, rt, intensity, files = seq.cuda(), charge.cuda(), rt.cuda(), intensity.cuda(), files.cuda()
pr_rt, pr_intensity = model.forward(seq, charge)
pred_rt.extend(pr_rt.data.cpu().tolist())
pred_int.extend(pr_intensity.data.cpu().tolist())
--
GitLab