TPU on Colab: unexpected keyword argument 'num_tpu_cores' #6244
Answered
by
akihironitta
naraugialusru
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
-
On a Colab TPU, I get the error:
I'm running a script that is in my Google Drive. I copy the data up onto Colab. When I do the exact same process for the GPU, it works (with Colab set to GPU, and the appropriate trainer parameter adjustment). The trainer instantiation is simply
|
Beta Was this translation helpful? Give feedback.
Answered by
akihironitta
Feb 28, 2021
Replies: 1 comment
-
@naraugialusru |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
naraugialusru
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@naraugialusru
num_tpu_cores
should betpu_cores
.num_tpu_cores
was removed in #2760. See also: the docs