Handle invalid momentum (inf) --kazu
pi0 script. cosmetic change in empart_training.py --kazu