more debug, hydra is not super friendly to debug, move to other logic for train fn next