refactor: rename pure_forward to reconstruct

Author: BerndDoser

src/spherinator/callbacks/log_reconstruction_callback.py

@@ -52,7 +52,7 @@ def on_train_epoch_end(self, trainer, model):
         images = images.to(model.device)
 
         # Generate reconstructions of the samples using the model
-        recon = model.pure_forward(images)
+        recon = model.reconstruct(images)
         loss = torch.nn.MSELoss(reduction="none")(images, recon).flatten(1).mean(1)
 
         # Plot the original samples and their reconstructions side by side

src/spherinator/models/autoencoder.py

@@ -38,9 +38,8 @@ def forward(self, x):
         x = self.encode(x)
         return self.decode(x)
 
-    def pure_forward(self, x):
-        x = self.encode(x)
-        return self.decode(x)
+    def reconstruct(self, x):
+        return self.forward(x)
 
     def _compute_loss(self, batch):
         recon = self.forward(batch)

src/spherinator/models/variational_autoencoder.py

@@ -118,7 +118,7 @@ def forward(self, x):
         recon = self.decode(z)
         return (z_location, z_scale), (q_z, p_z), z, recon
 
-    def pure_forward(self, x):
+    def reconstruct(self, x):
         z_location, _ = self.encode(x)
         return self.decode(z_location)