Removed old plots; fixed raster for paper

smsharma · Nov 28, 2023 · 19ade4b · 19ade4b
1 parent b2b2762
commit 19ade4b
Show file tree

Hide file tree

Showing 27 changed files with 107 additions and 102 deletions.
diff --git a/models/scores.py b/models/scores.py
@@ -9,7 +9,6 @@
 from models.transformer_adanorm import Transformer as TransformerAdaNorm
 from models.gnn import GraphConvNet
 from models.chebconv import ChebConvNet
-from models.dynamic_edge_conv import DynamicEdgeConvNet
 from models.mlp import MLP
 
 from models.graph_utils import nearest_neighbors, nearest_neighbors_kd, fourier_features
@@ -194,8 +193,6 @@ def __call__(self, z, t, conditioning, mask):
             h = jax.vmap(GraphConvNet(**score_dict, in_features=z.shape[-1]))(graph)
         elif self.gnn_type == "chebconv":
             h = jax.vmap(ChebConvNet(**score_dict))(graph)
-        elif self.gnn_type == "edgeconv":
-            h = jax.vmap(DynamicEdgeConvNet(**score_dict))(graph)
 
         # Predicted noise
         eps = graph.nodes - h.nodes

diff --git a/notebooks/03_viz_vel.ipynb b/notebooks/03_viz_vel.ipynb
diff --git a/notebooks/04_viz_pos.ipynb b/notebooks/04_viz_pos.ipynb
diff --git a/notebooks/plots/3d_animation.gif b/notebooks/plots/3d_animation.gif
diff --git a/notebooks/plots/3d_animation.mp4 b/notebooks/plots/3d_animation.mp4
diff --git a/notebooks/plots/animation.gif b/notebooks/plots/animation.gif
diff --git a/notebooks/plots/animation_both.gif b/notebooks/plots/animation_both.gif
diff --git a/notebooks/plots/animation_both_norotate.gif b/notebooks/plots/animation_both_norotate.gif
diff --git a/notebooks/plots/animation_diffusion.gif b/notebooks/plots/animation_diffusion.gif
diff --git a/notebooks/plots/animation_diffusion_jets.gif b/notebooks/plots/animation_diffusion_jets.gif
diff --git a/notebooks/plots/animation_diffusion_jets_mass.gif b/notebooks/plots/animation_diffusion_jets_mass.gif
diff --git a/notebooks/plots/animation_norotate.gif b/notebooks/plots/animation_norotate.gif
diff --git a/notebooks/plots/box_pos_viz.pdf b/notebooks/plots/box_pos_viz.pdf
diff --git a/notebooks/plots/box_vel_viz.pdf b/notebooks/plots/box_vel_viz.pdf
diff --git a/notebooks/plots/diff_viz.pdf b/notebooks/plots/diff_viz.pdf
diff --git a/notebooks/plots/diff_viz_ode.pdf b/notebooks/plots/diff_viz_ode.pdf
diff --git a/notebooks/plots/diff_viz_sde.pdf b/notebooks/plots/diff_viz_sde.pdf
diff --git a/notebooks/plots/diff_viz_sde_2.pdf b/notebooks/plots/diff_viz_sde_2.pdf
diff --git a/notebooks/plots/diff_viz_sde_rast.pdf b/notebooks/plots/diff_viz_sde_rast.pdf
diff --git a/notebooks/plots/gallant.pdf b/notebooks/plots/gallant.pdf
diff --git a/notebooks/plots/gallant_more.pdf b/notebooks/plots/gallant_more.pdf
diff --git a/notebooks/plots/leafy.pdf b/notebooks/plots/leafy.pdf
diff --git a/notebooks/plots/mnist_dark.png b/notebooks/plots/mnist_dark.png
diff --git a/notebooks/plots/qg.pdf b/notebooks/plots/qg.pdf
diff --git a/notebooks/plots/sequential_nbody.pdf b/notebooks/plots/sequential_nbody.pdf
diff --git a/notebooks/plots/t.pdf b/notebooks/plots/t.pdf
diff --git a/notebooks/plots/traj_cosmo.pdf b/notebooks/plots/traj_cosmo.pdf