NameError: name 'wasserstein_nms' is not defined

[Cathy1900]

你好，我训练atss_r50_aitod_nwd.py时，程序不报错。
但我训练detectors_cascade_rcnn_r50_aitod_rpn_nwd.py时，程序报错NameError: name 'wasserstein_nms' is not defined，请问是怎么回事呢？
环境如下：
Python: 3.7.15 (default, Nov 24 2022, 21:12:53) [GCC 11.2.0]
CUDA available: True
GPU 0,1: Tesla V100-SXM2-32GB
CUDA_HOME: /usr/local/cuda
NVCC: Cuda compilation tools, release 10.1, V10.1.105
GCC: gcc (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
PyTorch: 1.5.0+cu101
PyTorch compiling details: PyTorch built with:

• GCC 7.3
• C++ Version: 201402
• Intel(R) Math Kernel Library Version 2019.0.5 Product Build 20190808 for Intel(R) 64 architecture applications
• Intel(R) MKL-DNN v0.21.1 (Git Hash 7d2fd500bc78936d1d648ca713b901012f470dbc)
• OpenMP 201511 (a.k.a. OpenMP 4.5)
• NNPACK is enabled
• CPU capability usage: AVX2
• CUDA Runtime 10.1
• NVCC architecture flags: -gencode;arch=compute_37,code=sm_37;-gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_61,code=sm_61;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_75,code=sm_75;-gencode;arch=compute_37,code=compute_37
• CuDNN 7.6.3
• Magma 2.5.2
  TorchVision: 0.6.0+cu101
  OpenCV: 4.6.0
  MMCV: 1.3.5
  MMCV Compiler: GCC 7.3
  MMCV CUDA Compiler: 10.1
  MMDetection: 2.13.0+9775ac2

[fengchuibeixiang]

Have you solved the problem? I meet the same problem, could u please give me some advice? Thank u very much!