Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Closed
hydra-bu opened this issue May 14, 2024 · 2 comments
Closed

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

hydra-bu opened this issue May 14, 2024 · 2 comments

Comments

@hydra-bu
Copy link

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...

@hjh0119
Copy link
Collaborator

hjh0119 commented May 14, 2024

文档是源码版本的,对应版本是2.1.0.dev0
wheel包还没更新

@kratorado
Copy link

更新源码。

@hjh0119 hjh0119 closed this as completed May 20, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants