bi/embedding_model
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

qwen_train #2

Merged
alavi_team_user2 merged 4 commits from qwen_train into main 2025-11-23 10:54:27 +00:00
Conversation 0 Commits 4 Files Changed 4 +4 -1
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 7c945a1603 - Show all commits

1
.gitignore vendored
View File

@ -8,3 +8,4 @@ models
*.log *.log
research_notebook/data research_notebook/data
train/qwen/output train/qwen/output
train/qwen/mlruns

4
train/qwen/a.sh
View File

@ -3,6 +3,7 @@
nproc_per_node=1 nproc_per_node=1
MLFLOW_TRACKING_URI=http://0.0.0.0:5004 \
INFONCE_USE_BATCH=False \ INFONCE_USE_BATCH=False \
CUDA_VISIBLE_DEVICES=0 \ CUDA_VISIBLE_DEVICES=0 \
NPROC_PER_NODE=$nproc_per_node \ NPROC_PER_NODE=$nproc_per_node \
@ -30,4 +31,5 @@ swift sft \
--loss_type infonce \ --loss_type infonce \
--label_names labels \ --label_names labels \
--dataloader_drop_last true \ --dataloader_drop_last true \
--deepspeed zero3 --deepspeed zero3 \
--report_to mlflow