update repo

This commit is contained in:
嘉渊 2023-05-25 10:42:59 +08:00
parent 8e9b79cae3
commit ba4aef18ed

View File

@ -106,11 +106,12 @@ if __name__ == '__main__':
args.dataset, args.vocab_size, args.tag),
}
}
pipeline = {"type": args.pipeline_type}
json_dict = {
"framework": "pytorch",
"task": args.task,
"model": model,
"pipeline": args.pipeline_type,
"pipeline": pipeline,
}
with open(os.path.join(args.output_dir, "configuration.json"), "w") as f: