5fa1a76
1
2
3
4
5
6
7
yaml { "fp16": { "enabled": false } } For Ampere GPUs and PyTorch > 1.7, it automatically switches to the more efficient tf32 format for some operations but the results are still in fp32.
