release v0.6.0

Former-commit-id: 51910d5803eb718e4976da0b3bfcdc5eeeea48eb
This commit is contained in:
hiyouga
2024-03-25 22:38:56 +08:00
parent ac2a401b1d
commit daab85e3e6
5 changed files with 8 additions and 9 deletions

View File

@@ -416,7 +416,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