Fix GLUE TPU script
Temporary fix until we refactor this script to work with the trainer.
Pinging @jysohn23 to see if it fixed the issue.<|||||>Closing in favour of integrating TPU support for the trainer.