normalize

matthieutrs · matthieutrs · commit b70d651c58c5 · 2025-07-28T10:28:42.000+02:00
diff --git a/deepinv/models/ram.py b/deepinv/models/ram.py
@@ -217,7 +217,7 @@ def forward(self, y, physics=None, sigma=None, gain=None):
             )
 
         if physics is None:
-            gain = 1e-3 if gain is None else gain
+            gain = 1e-3 * y.abs().max() if gain is None else gain
             sigma = self.sigma_threshold if sigma is None else sigma
 
             physics = dinv.physics.Denoising(

Original file line number	Diff line number	Diff line change
`@@ -217,7 +217,7 @@ def forward(self, y, physics=None, sigma=None, gain=None):`
`217`	`217`	`)`
`218`	`218`
`219`	`219`	`if physics is None:`
`220`		`- gain = 1e-3 if gain is None else gain`
	`220`	`+ gain = 1e-3 * y.abs().max() if gain is None else gain`
`221`	`221`	`sigma = self.sigma_threshold if sigma is None else sigma`
`222`	`222`
`223`	`223`	`physics = dinv.physics.Denoising(`