Skip to content

Commit 3f95271

Browse files
authored
[RLFH] Fixed tensorboard bug (#10852)
1 parent 5c3a161 commit 3f95271

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

paddlenlp/trainer/integrations.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -198,14 +198,6 @@ def on_train_begin(self, args, state, control, **kwargs):
198198
if self.tb_writer is None:
199199
self._init_summary_writer(args, log_dir)
200200

201-
if self.tb_writer is not None:
202-
self.tb_writer.add_text("args", args.to_json_string())
203-
if "model" in kwargs:
204-
model = kwargs["model"]
205-
if hasattr(model, "config") and model.config is not None:
206-
model_config_json = model.config.to_json_string()
207-
self.tb_writer.add_text("model_config", model_config_json)
208-
209201
def on_log(self, args, state, control, logs=None, **kwargs):
210202
if not state.is_world_process_zero:
211203
return

0 commit comments

Comments
 (0)