Skip to content

Commit 01a0ee1

Browse files
authored
fix optimizer pararmeter issue (#7322)
Fixes # . ### Description A few sentences describing the changes proposed in this pull request. ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. Signed-off-by: binliu <[email protected]>
1 parent 4c546b3 commit 01a0ee1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/handlers/mlflow_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -401,7 +401,7 @@ def _default_iteration_log(self, engine: Engine) -> None:
401401
cur_optimizer = engine.optimizer
402402
for param_name in self.optimizer_param_names:
403403
params = {
404-
f"{param_name} group_{i}": float(param_group[param_name])
404+
f"{param_name}_group_{i}": float(param_group[param_name])
405405
for i, param_group in enumerate(cur_optimizer.param_groups)
406406
}
407407
self._log_metrics(params, step=engine.state.iteration)

0 commit comments

Comments
 (0)