Commit 81ba4c4c authored by Louis Martin's avatar Louis Martin Committed by Myle Ott
Browse files

Fix adaptive softmax cutoff comment

parent 47f095fd
...@@ -25,7 +25,7 @@ class AdaptiveSoftmax(nn.Module): ...@@ -25,7 +25,7 @@ class AdaptiveSoftmax(nn.Module):
cutoff = cutoff + [vocab_size] cutoff = cutoff + [vocab_size]
else: else:
assert vocab_size == cutoff[ assert vocab_size == cutoff[
-1], 'cannot specify cutoff smaller than vocab size' -1], 'cannot specify cutoff larger than vocab size'
output_dim = cutoff[0] + len(cutoff) - 1 output_dim = cutoff[0] + len(cutoff) - 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment