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):
cutoff = cutoff + [vocab_size]
else:
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
......
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