Skip to content

Commit 2ebb3dc

Browse files
blacksheep-Aristotlewaliwali777
authored andcommitted
update gpt run_pretrain_py
1 parent 303a193 commit 2ebb3dc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llm/auto_parallel/gpt-3/run_pretrain_auto.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -600,7 +600,7 @@ def fn(layer):
600600
checkpoint = training_args.resume_from_checkpoint
601601
elif last_checkpoint is not None:
602602
checkpoint = last_checkpoint
603-
print(trainer.model)
603+
604604
# Training
605605
if training_args.do_train:
606606
train_result = trainer.train(resume_from_checkpoint=checkpoint)

0 commit comments

Comments
 (0)