Skip to content

Commit 1f7612e

Browse files
[release/1.1] Fix protobuf 7 bug (PaddlePaddle#4120)
Co-authored-by: risemeup1 <62429225+risemeup1@users.noreply.github.com>
1 parent 1ae5268 commit 1f7612e

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

paddleformers/trainer/integrations.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,11 @@ def on_train_begin(self, args, state, control, **kwargs):
131131
# self.vdl_writer.add_text("model_config", model_config_json)
132132

133133
if hasattr(self.vdl_writer, "add_hparams"):
134-
self.vdl_writer.add_hparams(args.to_sanitized_dict(), metrics_list=[])
134+
# Convert bool to int for protobuf 7.x compatibility
135+
# protobuf 7.x is stricter about type validation and won't auto-convert bool to int
136+
sanitized_dict = args.to_sanitized_dict()
137+
sanitized_dict = {k: int(v) if isinstance(v, bool) else v for k, v in sanitized_dict.items()}
138+
self.vdl_writer.add_hparams(sanitized_dict, metrics_list=[])
135139

136140
def on_log(self, args, state, control, logs=None, **kwargs):
137141
if not state.is_world_process_zero:

0 commit comments

Comments
 (0)