use adaptive softmax only with adaptive loss

7f5f2461 · alexeib · Myle Ott · 7d560402 · 7f5f2461
Commit 7f5f2461 authored May 29, 2018 by alexeib Committed by Myle Ott Jun 15, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

fairseq/models/fconv.py fairseq/models/fconv.py +2 -1

No files found.
--- a/fairseq/models/fconv.py
+++ b/fairseq/models/fconv.py
@@ -139,7 +139,8 @@ class FConvLanguageModel(FairseqLanguageModel):
            max_positions=args.max_target_positions,
            share_embed=False,
            positional_embeddings=False,
-            adaptive_softmax_cutoff=options.eval_str_list(args.adaptive_softmax_cutoff, type=int),
+            adaptive_softmax_cutoff=options.eval_str_list(args.adaptive_softmax_cutoff,
+                                                          type=int) if args.criterion == 'adaptive_loss' else None,
            normalization_constant=args.normalization_constant,
        )
        return FConvLanguageModel(decoder)