Qwen Qlora ACSA
1.0.0
หน้าที่ของเราคือการกำหนดแนวโน้มทางอารมณ์ของแต่ละข้อความตรวจสอบร้านอาหารใน 18 มิติ 18 มิติเหล่านี้คือ:
หากคุณต้องการทราบข้อมูลเพิ่มเติมเกี่ยวกับชุดข้อมูลและตัวชี้วัดดู https://github.com/meituan-dianping/asap
อุปกรณ์ของฉัน: Linux, Pytorch2.0.1+Cu118, A100
mkdir -p /root/xtuner0117 && cd /root/xtuner0117
# Pull the source code of version 0.1.17
git clone -b v0.1.17 https://github.com/InternLM/xtuner
# Users who cannot access github please pull from gitee:
# git clone -b v0.1.15 https://gitee.com/Internlm/xtuner
# Enter the source code directory
cd /root/xtuner0117/xtuner
# Install XTuner from source
pip install -e '.[all]'
xtuner train qwen_1.8B_qlora_ASCA.py --deepspeed deepspeed_zero2 # Add deepspeed to accelerate training
xtuner convert pth_to_hf qwen_1.8B_qlora_ASCA.py
./work_dirs/qwen_1.8B_qlora_ASCA/iter_1803.pth ./hf
# Merge qlora files to generate fine-tuned qwen model
xtuner convert merge ./qwen/Qwen1.5-1.8B ./hf Qwen-1.5-1.8B-ASCA --max-shard-size 2GB
# Remove intermediate products
rm -rf ./hf
หากคุณต้องการทำการอนุมานคุณจะต้องดำเนินการ python main.py ในเทอร์มินัลระบบในไดเรกทอรีที่โฟลเดอร์ปัจจุบันอยู่ หากคุณกำหนดค่าสภาพแวดล้อมที่เกี่ยวข้องอย่างถูกต้องคุณจะทำงานได้สำเร็จ ความแม่นยำเฉลี่ยของชุดทดสอบทั้งหมดใน 18 มิติถึง 86.1%