Commit6bad8f6

authored

bugfix, remove default tensorboard writer to avoid permission issue

1 parentd0b8457 commit6bad8f6Copy full SHA for 6bad8f6

File tree

-4

lines changed

-4

lines changed

Lines changed: 0 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -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`	`defprint(self,msg:str):`
`156`	`155`	`"""`
`@@ -280,7 +279,6 @@ def accelerate_monitor(`
`280`	`279`
`281`	`280`	`ifself.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`	`ifselfpaced_statusisnotNone:`
`286`	`284`	`selfpaced_status.log_per_task_weight=torch.zeros(len(ID2TASK))`
`@@ -362,7 +360,6 @@ def accelerate_evaluate(`
`362`	`360`
`363`	`361`	`ifself.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`	`returneval_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}"`

Comments

(0)