def transformer_wmt_en_de_big_align(args): args.alignment_heads = getattr(args, "alignment_heads", 1) args.alignment_layer = getattr(args, "alignment_layer", 4) transformer_wmt_en_de_big(args)
def transformer_wmt_en_de_big_dist(args): transformer_wmt_en_de_big(args)
def position_transformer_wmt_en_de_big_align(args): # args.position_layers = getattr(args, "position_layers", [0, 5]) transformer_wmt_en_de_big(args)