OOMKilled with exit code 137 with finetune_trainer.py
Hi
I am trying to train on WMT with TPUs using seq2seq_trainer.py, and I always get "OOMKilled with exit code 137" on the line when it tries to use xm.spawn to run the main function on multiple TPUs. I decreased the batch size as much as possible, still the error is there. Do you have an idea why this is happening? thank you. I appreciate your help. 
I even set batch_size to 1 and still the error is there. any suggestion? <|||||>This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.
