Made train.py compatible with TensorFlow 2.0 and 1.0#9

rishiraj wants to merge 1 commit intoEMBLEBIGSOC:masterEMBLEBIGSOC/compara-deep-learning:masterfrom

rishiraj:patch-2rishiraj/compara-deep-learning:patch-2Copy head branch name to clipboard