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: