Forgot a minus and accidentally added the model

Author: miguelgondu

examples/black_box_random_geometries/von_mises_fisher_example/models/vmf_motion.pt

@@ -103,4 +103,4 @@ def run(
     print(model)
 
     # Trains
-    run(model, train_dataset, test_dataset, name="vmf_motion")
+    run(model, train_dataset, test_dataset, name="motion")

examples/black_box_random_geometries/von_mises_fisher_example/train.py

@@ -73,7 +73,7 @@ def forward(self, x: torch.Tensor):
         return q_z_given_x, p_x_given_z
 
     def elbo_loss(self, x: torch.Tensor, q_z_given_x: Normal, p_x_given_z: VonMisesFisher):
-        rec_loss = p_x_given_z.log_prob(x).sum(dim=1)
+        rec_loss = -p_x_given_z.log_prob(x).sum(dim=1)
         kl = kl_divergence(q_z_given_x, self.p_z).sum(dim=1)
 
         return (rec_loss + kl).mean()