Spaces:
Running
on
T4
Running
on
T4
noise
Browse files- factories.py +4 -4
factories.py
CHANGED
@@ -118,14 +118,14 @@ class PhysicsWithGenerator(torch.nn.Module):
|
|
118 |
circle=False,
|
119 |
normalize=True,
|
120 |
device=device_str,
|
121 |
-
noise_model=dinv.physics.GaussianNoise(sigma=1e-
|
122 |
max_iter=10,
|
123 |
)
|
124 |
-
self.physics_generator = SigmaGenerator(sigma_min=1e-
|
125 |
-
self.generator = SigmaGenerator(sigma_min=1e-
|
126 |
self.saved_params = {"updatable_params": {"sigma": 0.1},
|
127 |
"updatable_params_converter": {"sigma": float},
|
128 |
-
"fixed_params": {"noise_sigma_min": 1e-
|
129 |
"angles": angles, "max_iter": 10}}
|
130 |
|
131 |
def display_saved_params(self) -> str:
|
|
|
118 |
circle=False,
|
119 |
normalize=True,
|
120 |
device=device_str,
|
121 |
+
noise_model=dinv.physics.GaussianNoise(sigma=1e-6).to(device_str),
|
122 |
max_iter=10,
|
123 |
)
|
124 |
+
self.physics_generator = SigmaGenerator(sigma_min=1e-6, sigma_max=1e-6, device=device_str)
|
125 |
+
self.generator = SigmaGenerator(sigma_min=1e-6, sigma_max=1e-6, device=device_str)
|
126 |
self.saved_params = {"updatable_params": {"sigma": 0.1},
|
127 |
"updatable_params_converter": {"sigma": float},
|
128 |
+
"fixed_params": {"noise_sigma_min": 1e-6, "noise_sigma_max": 1e-6,
|
129 |
"angles": angles, "max_iter": 10}}
|
130 |
|
131 |
def display_saved_params(self) -> str:
|