Created by: peterzhang2029

For training script, there should be python train.py instead of python mt_with_external_memory.py