swift icon indicating copy to clipboard operation
swift copied to clipboard

自我认知 KeyError: 'self-cognition#500'

Open hydra-bu opened this issue 1 month ago • 2 comments

Describe the bug What the bug is, and how to reproduce, better with screenshots(描述bug以及复现过程,最好有截图)

使用UI 生成的命令行使用 --self_cognition_sample 参数(用这个参数是可以微调成功的),但是文档里又说该参数已经废弃,请使用--dataset , 使用新参数是报错的。

image

Your hardware and system info Write your system info like CUDA version/system/GPU/torch version here(在这里给出硬件信息和系统信息,如CUDA版本,系统,GPU型号和torch版本等)

Additional context Add any other context about the problem here(在这里补充其他信息)

执行的命令如下:

CUDA_VISIBLE_DEVICES=0 
swift sft \
    --sft_type lora \
    --model_id_or_path "AI-ModelScope/WizardLM-2-7B-AWQ" \
    --template_type "wizardlm2-awq" \
    --dataset alpaca-zh#500 alpaca-en#500 self-cognition#500 \
    --logging_steps 5 \
    --max_length 2048 \
    --learning_rate 5e-5 \
    --warmup_ratio 0.4 \
    --output_dir output \
    --lora_target_modules ALL \
    --model_name 小黄 'Xiao Huang' \
    --model_author 魔搭 ModelScope \

感觉文档内容更新不及时,照着敲也无法完成,指定自定义数据集也是各种报错,不是key_error就是column error...

hydra-bu avatar May 14 '24 08:05 hydra-bu