Spaces:
Sleeping
Sleeping
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:
|