cc

scripts/compute_latent.py

@@ -48,7 +48,7 @@ def compute_latent(model_name: str, swim_dir: str, batch_size: int):
             with torch.no_grad():
                 images = torch.stack(images).cuda()
                 latents = model.encode(images).latent_dist.mode()
-                latents = latents.clamp(-1, 1).detach().cpu().numpy()
+                latents = latents.detach().cpu().numpy()
 
             for name, latent in zip(image_names, latents):
                 torch.save(