Skip to content

GitLab

Commit 7f5f2461 authored by alexeib's avatar alexeib Committed by Myle Ott
Browse files

use adaptive softmax only with adaptive loss

parent 7d560402
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
fairseq/models/fconv.py
View file @ 7f5f2461
...@@ -139,7 +139,8 @@ class FConvLanguageModel(FairseqLanguageModel): ...@@ -139,7 +139,8 @@ class FConvLanguageModel(FairseqLanguageModel):
max_positions=args.max_target_positions, max_positions=args.max_target_positions,
share_embed=False, share_embed=False,
positional_embeddings=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, normalization_constant=args.normalization_constant,
) )
return FConvLanguageModel(decoder) return FConvLanguageModel(decoder)
......
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