Skip to content

Commit 0f150e5

Browse files
author
root
committed
Test wandb logging
1 parent 4c70ec2 commit 0f150e5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/special_e2e/sft/run_sft.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,6 @@ torchrun --standalone --nnodes=1 --nproc_per_node=${NUM_GPUS} ${ENTRYPOINT} \
5353
checkpoint.save_contents=[model,optimizer,extra,hf_model] \
5454
trainer.max_ckpt_to_keep=1 \
5555
trainer.resume_mode=${RESUME_MODE} \
56-
trainer.logger=['console'] $@
56+
trainer.logger=['console', 'wandb'] $@
5757

5858
rm -rf "${ckpts_home:?}/*"

0 commit comments

Comments
 (0)