feat: autoencoder + updated trainers + cleaned up process to allow using autoencoder

src/models/autoencoder/autoencoder.py

@@ -47,9 +47,15 @@ class AutoEncoder(Model):
         self.decoder = Decoder(latent_dim, channel_count, input_size)
 
     def encode(self, x: torch.Tensor) -> torch.Tensor:
+        """
+        x: torch.Tensor of floats
+        """
         return self.encoder(x)
 
     def decode(self, x: torch.Tensor) -> torch.Tensor:
+        """
+        x: torch.Tensor of floats
+        """
         return self.decoder(x)
 
     def forward(self, x: torch.LongTensor) -> torch.Tensor: