using fp16 or mixed precision help for inadequate vram

zouchuhang / LayoutNet

Torch implementation of our CVPR 18 paper: "LayoutNet: Reconstructing the 3D Room Layout from a Single RGB Image"

MIT License

417 stars 93 forks source link

Open parrykhai opened 5 years ago

parrykhai commented 5 years ago

Will mixed-precision be able to accommodate the 10.9gb in 8gb since using fp16 essentially doubles the VRAM.

zouchuhang commented 5 years ago

@parrykhai I haven't tried this. It's worthwhile to give it a try. If it works, you can send a new pull request and I will accept it.