From e57742b39ba5381dfe85f466f9d5fb4bcfca6fa5 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 12 Mar 2025 11:42:54 +0100
Subject: [PATCH] model cuda loading
---
dataset/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dataset/dataset.py b/dataset/dataset.py
index d64c917..baa79c3 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -11,7 +11,7 @@ class Threshold_noise:
self.threshold = threshold
def __call__(self, x):
- return torch.where((x <= self.threshold), torch.tensor(0.))
+ return torch.where((x <= self.threshold), 0.)
class Log_normalisation:
"""Log normalisation of intensities"""
--
GitLab