to below and retry.
elif FLAGS.mode.lower() == "multitask_classification" or FLAGS.mode.lower() == "classification":
to below and retry.
elif FLAGS.mode.lower() == "multitask_classification" or FLAGS.mode.lower() == "classification":