RuntimeError: floor_vml_cpu not implemented for ‘Int‘

在运行程序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类型向下取整导致的。

你可能感兴趣的:(python,深度学习,开发语言,pytorch)