PaddleNLP [问题]:使用uie-m-base进行训练时,max_seq_length设置为512报太小,设置为1024又爆显存,使用A100 40GB显存进行训练,

fdbelqdn  于 2个月前  发布在  其他
关注(0)|答案(2)|浏览(54)

请提出你的问题:
os.system('python finetune.py
--device "gpu"
--logging_steps 10
--save_steps 1000
--eval_steps 100
--seed 42
--model_name_or_path "./work/uie-m-base"
--output_dir ./work/model_best
--train_path ./data/train.txt
--dev_path ./data/dev.txt
--max_seq_length 512
--per_device_eval_batch_size 32
--per_device_train_batch_size 32
--num_train_epochs 20
--learning_rate 1e-5
--export_model_dir ./work/model_best
--overwrite_output_dir true
--disable_tqdm True
--metric_for_best_model eval_f1
--load_best_model_at_end True
--multilingual True
--do_train
--do_eval
--do_export
--save_total_limit 1')
pprint("训练完成")

rsaldnfx

rsaldnfx1#

请提供AIStudio的任务链接。

gajydyqb

gajydyqb2#

请提供AIStudio的任务链接
https://aistudio.baidu.com/projectdetail/8151763
我修改了代码,现在可以运行了

相关问题