add support for custom metric function for mixed precision#1420

Merged

Idan-BenAmi merged 6 commits intoSonySemiconductorSolutions:mainfrom

itai-berman:task_loss_api

Apr 28, 2025

Commits on Apr 22, 2025

add support for custom metric function for mixed precision
itai-berman
committed

Commits on Apr 23, 2025

remove duplicate custom metric function and keep only single entry point
itai-berman
committed
add custom_metric_fn description to docstring
itai-berman
committed

Commits on Apr 24, 2025

add check that returned type is float
itai-berman
committed

Commits on Apr 27, 2025

add unit test to mp custom metric
itai-berman
committed

Commits on Apr 28, 2025

move test to common and remove fixture
itai-berman
committed