suolyer commited on
Commit
887db86
·
1 Parent(s): 2839765

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -83,13 +83,13 @@ from fengshen.pipelines.multiplechoice import UniMCPipelines
83
  total_parser = argparse.ArgumentParser("TASK NAME")
84
  total_parser = UniMCPipelines.piplines_args(total_parser)
85
  args = total_parser.parse_args()
86
- args.pretrained_model_path = 'IDEA-CCNL/Erlangshen-UniMC-MegatronBERT-1.3B-Chinese'
87
  args.learning_rate=2e-5
88
  args.max_length=512
89
  args.max_epochs=3
90
  args.batchsize=8
91
  args.default_root_dir='./'
92
- model = UniMCPipelines(args)
93
 
94
  train_data = []
95
  dev_data = []
 
83
  total_parser = argparse.ArgumentParser("TASK NAME")
84
  total_parser = UniMCPipelines.piplines_args(total_parser)
85
  args = total_parser.parse_args()
86
+ pretrained_model_path = 'IDEA-CCNL/Erlangshen-UniMC-MegatronBERT-1.3B-Chinese'
87
  args.learning_rate=2e-5
88
  args.max_length=512
89
  args.max_epochs=3
90
  args.batchsize=8
91
  args.default_root_dir='./'
92
+ model = UniMCPipelines(args, pretrained_model_path)
93
 
94
  train_data = []
95
  dev_data = []