huang-yh / GaussianFormer

[ECCV 2024] Scene as Gaussians for Vision-Based 3D Semantic Occupancy Prediction
Other
278 stars 20 forks source link

Why use .cpu().inverse().cuda() ? #26

Closed HCaelrs closed 1 month ago

HCaelrs commented 1 month ago

Why not use .inverse() directly? https://github.com/huang-yh/GaussianFormer/blob/3bf28d50b41c8a54fda8147defe52a08f1a05b4e/model/head/gaussian_head.py#L94

huang-yh commented 1 month ago

We ran into errors when using inverse() directly on gpu. If it is ok on your machine, you can use inverse() directly.