We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2f80169 commit 7250150Copy full SHA for 7250150
1 file changed
src/maxtext/trainers/post_train/distillation/train_distill.py
@@ -228,7 +228,7 @@ def wrt_filter(path, x):
228
if not isinstance(x, base_wrt):
229
return False
230
freeze = student_freeze_param_filter(path)
231
- logging.info("Student model freezing info: Parameter {path}; freeze={freeze}")
+ logging.info(f"Student model freezing info: Parameter {path}; freeze={freeze}")
232
return not freeze
233
234
self.wrt_filter = wrt_filter
0 commit comments