Open Owen-Fish opened 2 years ago
For 1., according to the Decoder code:
def forward(self,source_fea,target_fea,mask):
# for x in self.decoder:
# print (x)
for i in range(self.log_size-2):
mask = transforms.Resize(source_fea[i].size(2))(mask)
blended = source_fea[i] * mask + target_fea[i] * (1 - mask)
if i == 0:
inputs = blended
else:
inputs = torch.cat([blended, outputs], dim=1)
#print (i,inputs.shape,self.decoder[i])
outputs = self.decoder[i](inputs)
return outputs
Thanks for your excellent work!
I recently saw this article on arxiv and would like to ask you for some details.
Looking for your reply, many thx!