月度归档: 2024 年 5 月

2 篇文章

thumbnail
CapAI炼丹记(3):写入自我认知
第一次尝试:失败,过拟合。 CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \ --stage sft \ --do_train True \ --model_name_or_path model/Baichuan2-13B-Base \ --adapter_name_or_path saves/Ba…
thumbnail
CapAI炼丹记(2):训练一个简单的对话模型2
接着上一篇文章CapAI炼丹记(1):训练一个简单的对话模型1 - FWERKOR博客,继续进行本地训练。 CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \ --stage sft \ --do_train True \ --model_name_or_path model/Baichuan2-13…