tusen-ai / SST

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).
Apache License 2.0
765 stars 96 forks source link

AssertionError: input should be in float32 type, got torch.float16 #181

Open lawrence-ff opened 5 months ago

lawrence-ff commented 5 months ago

微信图片_20240202181417 Hi, this problem is caused by the assertions in the norm.py file, it shows that some parts require the input data type to be torch.float32 but actually the input data type is torch.float16, what is causing this problem, is the interpreter not working,? looking forward to your reply, thanks!