Skip to content

Commit b0e2106

Browse files
authored
[cherry-pick][fix] internal_medicine_monitor_interval default to 0 (disabled) (#4751)
1 parent 73307aa commit b0e2106

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

paddleformers/trainer/trainer.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -496,11 +496,13 @@ def __init__(
496496
model, PipelineLayer
497497
), f"Only support pipeline parallel mode when model is PipelineLayer or PipelineLayer!!! but get {type(model.model)}"
498498
default_callbacks = DEFAULT_CALLBACKS.copy()
499-
if getattr(self.args, "internal_medicine_monitors", ""):
499+
_im_monitors = getattr(self.args, "internal_medicine_monitors", "")
500+
_im_interval = int(getattr(self.args, "internal_medicine_monitor_interval", 0))
501+
if _im_monitors and _im_interval > 0:
500502
default_callbacks.append(
501503
InternalMedicineCallback(
502-
monitors=self.args.internal_medicine_monitors,
503-
monitor_interval=self.args.internal_medicine_monitor_interval,
504+
monitors=_im_monitors,
505+
monitor_interval=_im_interval,
504506
)
505507
)
506508
default_callbacks += get_reporting_integration_callbacks(self.args.report_to)

0 commit comments

Comments
 (0)