wip prevent lattice squishing

curtischong · Apr 17, 2024 · f11e482 · f11e482
1 parent 466ce9b
commit f11e482
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/exploration/verify_vp_limited_mean_and_var.py b/exploration/verify_vp_limited_mean_and_var.py
@@ -1,6 +1,6 @@
 import pathlib
 from diffusion.diffusion_helpers import (
-    VP_limited_mean_and_var,
+    VP_lattice,
     polar_decomposition,
     symmetric_matrix_to_vector,
     vector_to_symmetric_matrix,
@@ -16,7 +16,7 @@
 # we want to sample many lattices at a high time step to see if the lattices look realistic
 def main():
     os.makedirs(OUT_DIR, exist_ok=True)
-    vp = VP_limited_mean_and_var(num_steps=1000, s=0.0001, power=2, clipmax=0.999)
+    vp = VP_lattice(num_steps=1000, s=0.0001, power=2, clipmax=0.999)  # noqa: F821
     t = 999  # sample from a very high time step for maximal variance
 
     square_lattice = torch.tensor(