mterris commited on
Commit
b2d938e
·
1 Parent(s): 3ac1bb3
Files changed (1) hide show
  1. 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-3).to(device_str),
122
  max_iter=10,
123
  )
124
- self.physics_generator = SigmaGenerator(sigma_min=1e-3, sigma_max=1e-3, device=device_str)
125
- self.generator = SigmaGenerator(sigma_min=1e-3, sigma_max=1e-3, 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-3, "noise_sigma_max": 1e-3,
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: