运行train.py，在第一个epoch报错

[stevenk12]

博主，您好，运行train.py时，在第一个epoch会报错： Traceback (most recent call last): File "", line 1, in File "D:\Anaconda3\envs\tensorflow2.4-gpu\lib\multiprocessing\spawn.py", line 105, in spawn_main exitcode = _main(fd) File "D:\Anaconda3\envs\tensorflow2.4-gpu\lib\multiprocessing\spawn.py", line 115, in _main self = reduction.pickle.load(from_parent) AttributeError: Can't get attribute 'TrainConfig' on <module 'mp_main' from 'f:\Algorithm\Algorithm_instance segmentation\mask-rcnn-tf2-master\mask-rcnn-tf2-master\train.py'>

我的配置是： CUDA11.0

scipy==1.4.1 numpy==1.19.2 matplotlib==3.2.1 opencv_python==4.2.0.34 tensorflow_gpu==2.4.0 tqdm==4.46.1 Pillow==9.2.0 h5py==2.10.0 scikit-image-0.16.2

[bubbliiiing]

多线程了？你设置了numworks？