From 5d2ff25ec76646b281c1144bd37ea7aee4f12cd4 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 12 Mar 2025 11:34:07 +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 6f807d0..551d448 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.max(x-self.threshold,0)
+ return torch.clamp(x, min=self.threshold)
class Log_normalisation:
"""Log normalisation of intensities"""
--
GitLab