RuntimeError: arguments are located on different GPUs

Would be better if you create an issue in simpletransformers repo.