From 48242dcb7f97d47db8dee294692a34553b81c1de Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 12 Mar 2025 11:39:40 +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 551d448..d64c917 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.clamp(x, min=self.threshold)
+ return torch.where((x <= self.threshold), torch.tensor(0.))
class Log_normalisation:
"""Log normalisation of intensities"""
--
GitLab