From ffdff12ccb436276dc0c0a64a3711a27074409cf Mon Sep 17 00:00:00 2001 From: Johannes Fischer Date: Tue, 20 Jul 2021 18:23:51 +0200 Subject: [PATCH] Update deepsets to deal with nans (first version) --- src/nets/deepsets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/nets/deepsets.py b/src/nets/deepsets.py index edbb4a0..f83d1b1 100644 --- a/src/nets/deepsets.py +++ b/src/nets/deepsets.py @@ -61,7 +61,7 @@ class DeepSetsModule(nn.Module): # use negative dynamic_dim since batch dimensions are inserted at the front dynamic_dim = -2 # iterate over dynamic dimension to apply phi to every instance - latent = tuple(self.phi(instance) for instance in x.unbind(dynamic_dim)) + latent = tuple(self.phi(instance) for instance in x.unbind(dynamic_dim) if torch.all(~torch.isnan(instance))) # stack outputs of phi latent = torch.stack(latent, dim=dynamic_dim) # apply pooling function to reduce dynamic dimension