Hi,
Recently, I looked at someone else’s code and found the following statement, but I did not find the corresponding change in the latest version of PL. I hope someone can help
self.to(self.trainer.accelerator_connector.root_gpu)
self.eval()
Hi,
Recently, I looked at someone else’s code and found the following statement, but I did not find the corresponding change in the latest version of PL. I hope someone can help
self.to(self.trainer.accelerator_connector.root_gpu)
self.eval()
hey @yipliu
you can access it using self.trainer.root_gpu
.
Also, we have moved the discussions to GitHub Discussions. You might want to check that out instead to get a quick response. The forums will be marked read-only after some time.
Thank you