search

rahul-goel / fused-ssim

Lightning fast differentiable SSIM.
MIT License
62 stars 3 forks source link

add device guard for multi GPU #2

Closed liruilong940607 closed 1 month ago

liruilong940607 commented 1 month ago

Fix for this issue https://github.com/nerfstudio-project/gsplat/issues/399#issuecomment-2348567118

I verified that with this fix, the multi-gpu script in gsplat can be executed without any issue.

rahul-goel commented 1 month ago

Thanks @liruilong940607 !