From b4f6e87395c1afe65cc59f2977741e5a24e42453 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sat, 25 May 2024 21:40:21 +0200 Subject: [PATCH] Remove pdb --- models.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/models.py b/models.py index caadb72..a84faa9 100644 --- a/models.py +++ b/models.py @@ -51,8 +51,6 @@ class MiddleOut(nn.Module): new_latents = [] for p in range(peer_latents.shape[-2]): peer_latent, correlation = peer_latents[:, p, :], peer_correlations[:, p] - import pdb - pdb.set_trace() combined_input = torch.cat((my_latent, peer_latent, correlation.unsqueeze(1)), dim=-1) new_latent = self.fc(combined_input) new_latents.append(new_latent * correlation.unsqueeze(1))