[script] Documented --epochs on both train/experiment scripts to encompass the tip on the doc of train