Skip to content
Snippets Groups Projects
Commit 44a88821 authored by Schneider Leo's avatar Schneider Leo
Browse files

model cuda loading

parent e57742b3
No related merge requests found
...@@ -11,7 +11,7 @@ class Threshold_noise: ...@@ -11,7 +11,7 @@ class Threshold_noise:
self.threshold = threshold self.threshold = threshold
def __call__(self, x): def __call__(self, x):
return torch.where((x <= self.threshold), 0.) return torch.where((x <= self.threshold), 0.,x)
class Log_normalisation: class Log_normalisation:
"""Log normalisation of intensities""" """Log normalisation of intensities"""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment