Add `train.py` and `val.py` callbacks (#4220) b74929c unverified Kalen Michael glenn-jocher commited on Jul 31, 2021