release v0.6.0

This commit is contained in:
hiyouga
2024-03-25 22:38:56 +08:00
parent f33a3dfadc
commit 6f2b563f12
5 changed files with 8 additions and 9 deletions

View File

@@ -415,7 +415,7 @@ CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
--lr_scheduler_type cosine \
--logging_steps 10 \
--save_steps 1000 \
--learning_rate 1e-6 \
--learning_rate 1e-5 \
--num_train_epochs 1.0 \
--plot_loss \
--fp16