Draft: Fix class imbalance in CE loss#56

Draft

PatrickRMiles wants to merge 6 commits intoLBANN:mainfrom

PatrickRMiles:miles30/class_imbalance

Commits on Apr 22, 2026

Calculate local CE loss w/o AMP to prevent inf from f16 overflow
Patrick Miles
committed
ruff
Patrick Miles
committed
use class weights in CE loss to make background less dominant; calc weights at trainer init
Patrick Miles
committed
ruff
Patrick Miles
committed
missing import
Patrick Miles
committed
ruff
Patrick Miles
committed