代码在CPU上运行 - Githubissues

haofeixu / aanet

[CVPR'20] AANet: Adaptive Aggregation Network for Efficient Stereo Matching

Apache License 2.0

524 stars 102 forks source link

Closed wangzhao88 closed 4 years ago

wangzhao88 commented 4 years ago

您好，请问这份代码是否有在纯CPU上运行过呢？

haofeixu commented 4 years ago

The deformable conv op is supported by CUDA only. If you want to run on CPU, you may disable all deformable convs in our model and retrain yourself.

wangzhao88 commented 4 years ago

OK, thank you.