Faire évoluer la précision des opérations matricielles

IGNF / myria3d

Myria3D: Aerial Lidar HD Semantic Segmentation with Deep Learning

https://ignf.github.io/myria3d/

BSD 3-Clause "New" or "Revised" License

151 stars 20 forks source link

Faire évoluer la précision des opérations matricielles #110

Open CharlesGaydon opened 5 months ago

CharlesGaydon commented 5 months ago

You are using a CUDA device ('NVIDIA A40') that has Tensor Cores. To properly utilize them, you should set 
`torch.set_float32_matmul_precision('medium' | 'high')` which will trade-off precision for performance. For more details, read 
https://pytorch.org/docs/stable/generated/torch.set_float32_matmul_precision.html#torch.set_float32_matmul_precision

-> évaluation du gain à effectuer.