Fix model size mismatch on restart#9

Merged

michaelmckinsey1 merged 3 commits intoLBANN:mainfrom

PatrickRMiles:fix/miles30/restart_size_mismatch

Jan 29, 2026

Commits on Jan 29, 2026

fix cli bug: must recalculate unet_layers in CLI since problem_scale can be overwritten iwhtout changing config object unet_layers
Patrick Miles
committed
whitespace
Patrick Miles
committed
ruff
Patrick Miles
committed