Updates on examples

[97littleleaf11]

This PR does some minor fixes to the examples:

• Move the to(device) commands into dataloader
• Remove redundance check of steps > 0
• Add a --data_root config and remove the hard-encoded path

[liruilong940607]

Hey thanks for cleaning them up!