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

Fix adaptive softmax cutoff comment

parent 47f095fd

fairseq/modules/adaptive_softmax.py

+1 −1
Original line number Diff line number Diff line
@@ -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