I have some cutom metrics that I need to monitor during training. I already implemented the the functions and they work fine but I can\'t seem to figure out how to assign na
