Spaces:
Runtime error
Runtime error
Commit
·
55360ff
1
Parent(s):
73933e0
add logging
Browse files
app.py
CHANGED
@@ -47,6 +47,7 @@ def respond(custom_string):
|
|
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(out), 1, p=out)
|
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(out)
|
51 |
ix = np.random.choice(np.arange(out), 1, p=out)
|
52 |
# val, ix = out[:, -1].data.topk(1)
|
53 |
|