Commit fa508492 authored by Myle Ott's avatar Myle Ott
Browse files

Output number of model parameters in train.py

parent 18a6d85c

train.py

+1 −0
Changes for train.py: 1 added line, 0 removed lines.
Original line number Diff line number Diff line
@@ -70,6 +70,7 @@ def main(): 
    model = utils.build_model(args, dataset.src_dict, dataset.dst_dict)

    criterion = utils.build_criterion(args, dataset.src_dict, dataset.dst_dict)

    print('| model {}, criterion {}'.format(args.arch, criterion.__class__.__name__))

    print('| num. model params: {}'.format(sum(p.data.numel() for p in model.parameters())))



    # The max number of positions can be different for train and valid

    # e.g., RNNs may support more positions at test time than seen in training