init got an unexpected keyword argument gpus

Question

There is an error which i got that is __init__() got an unexpected keyword argument 'gpus'

Answer 1

This error typically happens when using PyTorch Lightning—replace gpus with devices and accelerator='gpu' for newer versions.

Here is the code snippet you can refer to:

In the above code we are using the following key points:

• Uses accelerator='gpu' instead of gpus for PyTorch Lightning compatibility.
• Sets devices=1 to specify the number of GPUs.
• Ensures model training on GPU with updated syntax.

 Hence, using accelerator='gpu' and devices=1 resolves the unexpected keyword argument error.