I had problems running the model on gpu, since my gpu only has 6GB of memory. When I tried to run it on cpu, I received an error about amp not being compatible with cpu. This pull request changes the pl.Trainer parameters if the run will happen on cpu.
Hi,
I had problems running the model on gpu, since my gpu only has 6GB of memory. When I tried to run it on cpu, I received an error about amp not being compatible with cpu. This pull request changes the pl.Trainer parameters if the run will happen on cpu.
Greetings, Christian