Skip to content

Commit

Permalink
Type error
Browse files Browse the repository at this point in the history
It seems the support variable is a long not a float
manuel-delverme authored Mar 5, 2020
1 parent c7d289b commit 3edc1d2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion self_play.py
Original file line number Diff line number Diff line change
@@ -331,7 +331,7 @@ def support_to_scalar(logits, support_size):
torch.tensor([x for x in range(-support_size, support_size + 1)])
.expand(probs.shape)
.to(device=probs.device)
)
).float()
x = torch.sum(support * probs, dim=1, keepdim=True)

# Invert the scaling (defined in https://arxiv.org/abs/1805.11593)

0 comments on commit 3edc1d2

Please sign in to comment.