Fix slow gpu start and rewrite csph3d model

felipegb94 / learned-compressive-spad-histograms-iccv2023

Code for ICCV 2023 paper: Learned compressive representations for single-photon 3d imaging

https://wisionlab.com/project/learned-compressive-representations-for-single-photon-3d-cameras/

MIT License

3 stars 0 forks source link

Fix slow gpu start and rewrite csph3d model #1

Closed felipegb94 closed 1 year ago

felipegb94 commented 1 year ago

Set cudnn backend benchmark flag to False. This makes sure that there train/test does not take a long time in the first forward pass. I did not find any significant performance hits.

I also re-wrote the CSPH3DLayer. The old CSPH3D implementation was now renamed to be CSPH3DLayerv1 , and all the necessary files were modified to propagate that change so that we can continue to use the models that were trained with CSPH3DLayerv1 (previous implementatiion).