From 40dc2938e88cf3faa72ec8e56bf7ddbcf0c4f6af Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 22 Oct 2020 13:27:06 -0600 Subject: [PATCH] Fix multifaceted chain gen --- codes/models/archs/ChainedEmbeddingGen.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/codes/models/archs/ChainedEmbeddingGen.py b/codes/models/archs/ChainedEmbeddingGen.py index 45a267cc..35366841 100644 --- a/codes/models/archs/ChainedEmbeddingGen.py +++ b/codes/models/archs/ChainedEmbeddingGen.py @@ -226,8 +226,7 @@ class MultifacetedChainedEmbeddingGen(nn.Module): # Integrate recurrence inputs. if teco_recurrent is not None: - teco_rec = torch.nn.functional.interpolate(teco_recurrent, scale_factor=2, mode='nearest') - teco_rec = self.teco_recurrent_process(teco_rec) + teco_rec = self.teco_recurrent_process(teco_recurrent) fea, std = self.teco_recurrent_join(fea, teco_rec) self.teco_ref_std = std.item() elif prog_recurrent is not None: