There exist different possibilties to set at least some operations to fp16 The one using bf16-mixed at the configfile for LigthningCLI does not work at the moment... Tracking of one problem at: https://github.com/pytorch/pytorch/issues/112575
There exist different possibilties to set at least some operations to fp16
The one using bf16-mixed at the configfile for LigthningCLI does not work at the moment...
Tracking of one problem at: pytorch/pytorch#112575