diff --git a/mm-zero3.sh b/mm-zero3.sh index aaf6807..ca051f5 100755 --- a/mm-zero3.sh +++ b/mm-zero3.sh @@ -40,6 +40,6 @@ deepspeed --hostfile hostfile \ --gradient_checkpointing \ --bf16 \ --deepspeed /home/test/jd_train/ds_config_zero3.json \ - --report_to none \ + --report_to wandb \ --eval_data_glob "/home/test/datasets/my_corpus/test.jsonl"