在运行程序MinkowskiEngine时,遇到一个问题:
File "/home/yons/anaconda3/envs/GSNet/lib/python3.7/site-packages/MinkowskiEngine-0.5.4-py3.7-linux-x86_64.egg/MinkowskiEngine/utils/quantization.py", line 133, in _auto_floor
return torch.floor(array)
RuntimeError: floor_vml_cpu not implemented for 'Int'
网上看到的解决方法很少,打印type(array)发现是dtype=torch.int32。
其实只需要加一个判断:
if array.dtype == torch.int32:
return array
else:
return torch.floor(array)
该问题可能是由于torch.floor无法对int32类型向下取整导致的。