KeyError: "Registering two shape functions with name 'NnDistance' !

真是見鬼,編譯tf_nndistance_g.cu這些C++ OP都沒問題,但是調用就出現了這個問題:KeyError: "Registering two shape functions with name ‘NnDistance’ !。浪費了我好幾天,擦!!!
解決方法:
在這裏插入圖片描述上面的structural_losses文件夾進行mark source root的話,就需要把structural_losses這個刪了。在這裏插入圖片描述

如果編譯過程沒問題的話,出現類似上訴問題都可以這樣解決。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章