Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Commit7898fa6

Browse files
committed
remove default writer
1 parente827775 commit7898fa6

File tree

2 files changed

+0
-8
lines changed

2 files changed

+0
-8
lines changed

‎mftcoder_accelerate/src/mpt/mpt_trainer.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,6 @@ def __init__(
147147
self.args=args
148148
# tensorboard writer
149149
self.summary_writer=SummaryWriter(log_dir=args.tb_dir)
150-
self.default_writer=SummaryWriter(log_dir="/home/admin/logs/tfevent")
151150

152151
defprint(self,msg:str):
153152
"""
@@ -282,7 +281,6 @@ def accelerate_monitor(
282281

283282
ifself.accelerator.is_main_process:
284283
write_tensorboard(self.summary_writer,train_log_dict,completed_steps)
285-
write_tensorboard(self.default_writer,train_log_dict,completed_steps)
286284

287285
ifselfpaced_statusisnotNone:
288286
selfpaced_status.log_per_task_weight=torch.zeros(len(ID2TASK))
@@ -364,7 +362,6 @@ def accelerate_evaluate(
364362

365363
ifself.accelerator.is_main_process:
366364
write_tensorboard(self.summary_writer,eval_log_dict,completed_steps)
367-
write_tensorboard(self.default_writer,eval_log_dict,completed_steps)
368365

369366
returneval_loss,eval_task_loss,min_eval_loss,stall_num,best_step
370367

@@ -593,4 +590,3 @@ def accelerate_train(self):
593590
self.accelerate_saving_states(output_dir,completed_steps)
594591

595592
self.summary_writer.close()
596-
self.default_writer.close()

‎mftcoder_accelerate/src/pefts/mft_trainer.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,6 @@ def __init__(
150150
self.args=args
151151
# tensorboard writer
152152
self.summary_writer=SummaryWriter(log_dir=args.tb_dir)
153-
self.default_writer=SummaryWriter(log_dir="/home/admin/logs/tfevent")
154153

155154
defprint(self,msg:str):
156155
"""
@@ -278,7 +277,6 @@ def accelerate_monitor(
278277

279278
ifself.accelerator.is_main_process:
280279
write_tensorboard(self.summary_writer,train_log_dict,completed_steps)
281-
write_tensorboard(self.default_writer,train_log_dict,completed_steps)
282280

283281
ifselfpaced_statusisnotNone:
284282
selfpaced_status.log_per_task_weight=torch.zeros(len(ID2TASK))
@@ -360,7 +358,6 @@ def accelerate_evaluate(
360358

361359
ifself.accelerator.is_main_process:
362360
write_tensorboard(self.summary_writer,eval_log_dict,completed_steps)
363-
write_tensorboard(self.default_writer,eval_log_dict,completed_steps)
364361

365362
returneval_loss,eval_task_loss,min_eval_loss,stall_num,best_step
366363

@@ -587,7 +584,6 @@ def accelerate_train(self):
587584
self.accelerate_saving_checkpoint(output_dir,completed_steps)
588585

589586
self.summary_writer.close()
590-
self.default_writer.close()
591587

592588
# final save
593589
# output_dir = f"final_step_{completed_steps}"

0 commit comments

Comments
 (0)

[8]ページ先頭

©2009-2025 Movatter.jp