nickgardner commited on
Commit
6dcbe54
·
1 Parent(s): 3a3549f
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -47,6 +47,7 @@ def respond(input):
47
 
48
  out = model.out(model.decoder(outputs[:i].unsqueeze(0), e_outputs, src_mask, trg_mask))
49
  out = torch.nn.functional.softmax(out, dim=-1).detach()
 
50
  ix = np.random.choice(np.arange(len(out[:, -1].data[0])), 1, p=out[:, -1].data[0])
51
  # val, ix = out[:, -1].data.topk(1)
52
 
 
47
 
48
  out = model.out(model.decoder(outputs[:i].unsqueeze(0), e_outputs, src_mask, trg_mask))
49
  out = torch.nn.functional.softmax(out, dim=-1).detach()
50
+ print(np.sum(out[:, -1].data[0]))
51
  ix = np.random.choice(np.arange(len(out[:, -1].data[0])), 1, p=out[:, -1].data[0])
52
  # val, ix = out[:, -1].data.topk(1)
53