(args)
| 140 | |
| 141 | |
| 142 | def main(args): |
| 143 | if args.src_seq_length > args.max_position_embeddings: |
| 144 | args.max_position_embeddings = args.src_seq_length |
| 145 | if args.task.lower() in ['cnn_dm', 'cnn_dm_original', 'gigaword', 'blank', 'squad_generation', 'xsum', |
| 146 | 'squad', 'squad_v1', 'extraction', 'cmrc', 'customization']: |
| 147 | finetune(args, train_valid_datasets_provider, {}, end_of_epoch_callback_provider=metrics_func_provider, |
| 148 | forward_step=seq2seq_forward_step) |
| 149 | else: |
| 150 | raise NotImplementedError(args.task) |
no test coverage detected