search

NVlabs / VoxFormer

Official PyTorch implementation of VoxFormer [CVPR 2023 Highlight]
Other
1.07k stars 87 forks source link

how to train the VoxFormer at precision of fp16? #60

Open hitbuyi opened 4 months ago

hitbuyi commented 4 months ago

Is there any example to demostrate training the voxformer in fp16?

Shiyao-Xu commented 4 months ago

I guess you can set @auto_fp16 where you need it and make sure that your gpu supports fp16 computation.

hitbuyi commented 4 months ago

I guess you can set @auto_fp16 where you need it and make sure that your gpu supports fp16 computation.

I will have a try