Why do you need "force_fp32"?

thangvubk / SoftGroup

[CVPR 2022 Oral] SoftGroup for Instance Segmentation on 3D Point Clouds

MIT License

339 stars 80 forks source link

Closed Pixie8888 closed 8 months ago

Pixie8888 commented 8 months ago

Hi,

I notice the function force_fp32 in softgroup/util/fp16.py. Why do you need to force it to be fp32 when training with amp?

thangvubk commented 8 months ago

It is implemented for mix precision training. But it is not in used currently.

Pixie8888 commented 8 months ago

Thanks