Closed AntonioMirarchi closed 1 year ago
If you see no performance regression when training with several GPUs then its great!
@AntonioMirarchi is this ready to merge?
This is solving the warning only with single GPU training, while for multiple-GPU training is still there
I merged this one, thanks Antonio!
When you train on a single GPU the following warning raise up:
In order to remove this warning I'm proposing in this PR to change the strategy used in the
pl.Trainer()
to "auto". Another way could be to set the strategy based on the GPUs involved in the training.