Skip to content

Commit 6bad8f6

Browse files
authored
bugfix, remove default tensorboard writer to avoid permission issue
1 parent d0b8457 commit 6bad8f6

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

mftcoder_accelerate/src/pefts/trainer.py

-4
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
def print(self, msg: str):
156155
"""
@@ -280,7 +279,6 @@ def accelerate_monitor(
280279

281280
if self.accelerator.is_main_process:
282281
write_tensorboard(self.summary_writer, train_log_dict, completed_steps)
283-
write_tensorboard(self.default_writer, train_log_dict, completed_steps)
284282

285283
if selfpaced_status is not None:
286284
selfpaced_status.log_per_task_weight = torch.zeros(len(ID2TASK))
@@ -362,7 +360,6 @@ def accelerate_evaluate(
362360

363361
if self.accelerator.is_main_process:
364362
write_tensorboard(self.summary_writer, eval_log_dict, completed_steps)
365-
write_tensorboard(self.default_writer, eval_log_dict, completed_steps)
366363

367364
return eval_loss, eval_task_loss, min_eval_loss, stall_num, best_step
368365

@@ -589,7 +586,6 @@ def accelerate_train(self):
589586
self.accelerate_saving_checkpoint(output_dir, completed_steps)
590587

591588
self.summary_writer.close()
592-
self.default_writer.close()
593589

594590
# final save
595591
# output_dir = f"final_step_{completed_steps}"

0 commit comments

Comments
 (0)