From 44a88821c2f052c424638c3b419ebc92f94de375 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 12 Mar 2025 11:45:27 +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 baa79c3..446d281 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), 0.)
+ return torch.where((x <= self.threshold), 0.,x)
class Log_normalisation:
"""Log normalisation of intensities"""
--
GitLab