Run bert_score on TPU (google colab) instead of gpu.

[seyyedjavadrazavi]

trafficstars

Hi there. Is there any way to run bert_score on tpu? If yes, please explain or send a straightforward link.

I'd like to use the score function like below: score(input_txt, sim_txt, lang="en", model_type = "lanwuwei/BERTOverflow_stackoverflow_github", num_layers = steps, device="cuda:0", nthreads=4, rescale_with_baseline=True)

But on the TPU, and I don't have any idea about that. Thanks in advance.