clean up train.py for clarity and flexibility

[bdecost]

we've ended up with a lot of complex logic in the main training script -- I think we should consider moving ignite event handler definitions out of the main train_dgl body and register handlers with add_event_handler instead of the @on decorator trainer.add_event_handler(Events.STARTED, lambda _: print("Start training"))