public-projects/neuro-symbolic-visual-dialog
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Update 'prog_generator/models.py'

Browse source
This commit is contained in:
Adnen Abdessaied 2023-03-20 16:07:24 +01:00
parent 6de837ea53
commit ce9c902570
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
prog_generator/models.py
View file

@ -456,6 +456,12 @@ class SeqToSeqC(nn.Module):
predSoftmax, progHC = self.decoder(prog, encOut, capO) predSoftmax, progHC = self.decoder(prog, encOut, capO)
return predSoftmax, progHC return predSoftmax, progHC
def sample(self, cap):
with torch.no_grad():
encOut, capO = self.encoder(cap)
outputTokens, outputLogProbs = self.decoder.sample(encOut, capO)
outputTokens = torch.stack(outputTokens, 0).transpose(0, 1)
return outputTokens
class SeqToSeqQ(nn.Module): class SeqToSeqQ(nn.Module):
def __init__(self, encoder, decoder): def __init__(self, encoder, decoder):