2d52dc9292ffcd084125507241db6170e18ac629,main.py,,main,#Any#,90

Before Change

        val_results, _ = evaluate(model, tasks, args.batch_size, args.cuda, "val")
        if args.write_preds:
            if len(tasks) == 1 and isinstance(tasks[0], NLITypeProbingTask):
                _, te_preds = evaluate(model, tasks, args.batch_size, args.cuda, "val")
            else:
                _, te_preds = evaluate(model, tasks, args.batch_size, args.cuda, "test")
            write_preds(te_preds, args.run_dir)

After Change

        if "val" in splits_to_write:
            evaluate.write_preds(val_preds, args.run_dir, "val",
                                 strict_glue_format=args.write_strict_glue_format)
        if "test" in splits_to_write:
            _, te_preds = evaluate.evaluate(model, tasks,
                                            args.batch_size, args.cuda, "test")
            evaluate.write_preds(te_preds, args.run_dir, "test",
                                 strict_glue_format=args.write_strict_glue_format)
        run_name = args.get("run_name", os.path.basename(args.run_dir))
        evaluate.write_results(val_results, os.path.join(args.exp_dir, "results.tsv"),
                               run_name=run_name)

Instances