Merge pull request #61 from codefuse-ai/chencyudel-patch-2

bugfix, remove default tensorboard writer to avoid permission issue

Author: chencyudel

mftcoder_accelerate/src/pefts/trainer.py

@@ -150,7 +150,6 @@ def __init__(
         self.args = args
         # tensorboard writer
         self.summary_writer = SummaryWriter(log_dir=args.tb_dir)
-        self.default_writer = SummaryWriter(log_dir="/home/admin/logs/tfevent")
 
     def print(self, msg: str):
         """
@@ -280,7 +279,6 @@ def accelerate_monitor(
 
         if self.accelerator.is_main_process:
             write_tensorboard(self.summary_writer, train_log_dict, completed_steps)
-            write_tensorboard(self.default_writer, train_log_dict, completed_steps)
 
         if selfpaced_status is not None:
             selfpaced_status.log_per_task_weight = torch.zeros(len(ID2TASK))
@@ -362,7 +360,6 @@ def accelerate_evaluate(
 
         if self.accelerator.is_main_process:
             write_tensorboard(self.summary_writer, eval_log_dict, completed_steps)
-            write_tensorboard(self.default_writer, eval_log_dict, completed_steps)
 
         return eval_loss, eval_task_loss, min_eval_loss, stall_num, best_step
 
@@ -589,7 +586,6 @@ def accelerate_train(self):
                 self.accelerate_saving_checkpoint(output_dir, completed_steps)
 
         self.summary_writer.close()
-        self.default_writer.close()
 
         # final save
         # output_dir = f"final_step_{completed_steps}"