@@ -150,7 +150,6 @@ def __init__(
150
150
self .args = args
151
151
# tensorboard writer
152
152
self .summary_writer = SummaryWriter (log_dir = args .tb_dir )
153
- self .default_writer = SummaryWriter (log_dir = "/home/admin/logs/tfevent" )
154
153
155
154
def print (self , msg : str ):
156
155
"""
@@ -280,7 +279,6 @@ def accelerate_monitor(
280
279
281
280
if self .accelerator .is_main_process :
282
281
write_tensorboard (self .summary_writer , train_log_dict , completed_steps )
283
- write_tensorboard (self .default_writer , train_log_dict , completed_steps )
284
282
285
283
if selfpaced_status is not None :
286
284
selfpaced_status .log_per_task_weight = torch .zeros (len (ID2TASK ))
@@ -362,7 +360,6 @@ def accelerate_evaluate(
362
360
363
361
if self .accelerator .is_main_process :
364
362
write_tensorboard (self .summary_writer , eval_log_dict , completed_steps )
365
- write_tensorboard (self .default_writer , eval_log_dict , completed_steps )
366
363
367
364
return eval_loss , eval_task_loss , min_eval_loss , stall_num , best_step
368
365
@@ -589,7 +586,6 @@ def accelerate_train(self):
589
586
self .accelerate_saving_checkpoint (output_dir , completed_steps )
590
587
591
588
self .summary_writer .close ()
592
- self .default_writer .close ()
593
589
594
590
# final save
595
591
# output_dir = f"final_step_{completed_steps}"
0 commit comments