真是見鬼,編譯tf_nndistance_g.cu這些C++ OP都沒問題,但是調用就出現了這個問題:KeyError: "Registering two shape functions with name ‘NnDistance’ !。浪費了我好幾天,擦!!!
解決方法:
上面的structural_losses文件夾進行mark source root的話,就需要把structural_losses這個刪了。
如果編譯過程沒問題的話,出現類似上訴問題都可以這樣解決。
真是見鬼,編譯tf_nndistance_g.cu這些C++ OP都沒問題,但是調用就出現了這個問題:KeyError: "Registering two shape functions with name ‘NnDistance’ !。浪費了我好幾天,擦!!!
解決方法:
上面的structural_losses文件夾進行mark source root的話,就需要把structural_losses這個刪了。
如果編譯過程沒問題的話,出現類似上訴問題都可以這樣解決。