dodox/Spikey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove pdb

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-05-25 21:40:21 +02:00
parent b4fe95ef6c
commit b4f6e87395
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -51,8 +51,6 @@ class MiddleOut(nn.Module):
new_latents = [] new_latents = []
for p in range(peer_latents.shape[-2]): for p in range(peer_latents.shape[-2]):
peer_latent, correlation = peer_latents[:, p, :], peer_correlations[:, p] 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) combined_input = torch.cat((my_latent, peer_latent, correlation.unsqueeze(1)), dim=-1)
new_latent = self.fc(combined_input) new_latent = self.fc(combined_input)
new_latents.append(new_latent * correlation.unsqueeze(1)) new_latents.append(new_latent * correlation.unsqueeze(1))