真是见鬼,编译tf_nndistance_g.cu这些C++ OP都没问题,但是调用就出现了这个问题:KeyError: "Registering two shape functions with name ‘NnDistance’ !。浪费了我好几天,擦!!!
解决方法:
KeyError: "Registering two shape functions with name 'NnDistance' !上面的structural_losses文件夹进行mark source root的话,就需要把structural_losses这个删了。KeyError: "Registering two shape functions with name 'NnDistance' !

如果编译过程没问题的话,出现类似上诉问题都可以这样解决。

相关文章:

  • 2022-12-23
  • 2022-12-23
  • 2021-09-19
  • 2022-01-07
  • 2022-01-03
  • 2021-09-11
猜你喜欢
  • 2022-01-01
  • 2021-11-05
  • 2021-06-22
  • 2021-08-19
  • 2021-09-05
  • 2022-12-23
  • 2021-11-21
相关资源
相似解决方案