在做 https://github.com/matterport/Mask_RCNN 的复现。
在复现train_shpes时,在heads层训练时,卡在了Epoch 1/1。我观察下gpu和cpu,都没有工作
我在停止代码运行时发现停在了
File "", line 1, in
runfile('D:/py/Mask_RCNN-master/samples/shapes/train_shapes.py', wdir='D:/py/Mask_RCNN-master/samples/shapes')
File "D:\anaconda\envs\tensorflow\lib\site-packages\spyder\utils\site\sitecustomize.py", line 705, in runfile
execfile(filename, namespace)
File "D:\anaconda\envs\tensorflow\lib\site-packages\spyder\utils\site\sitecustomize.py", line 102, in execfile
exec(compile(f.read(), filename, 'exec'), namespace)
File "D:/py/Mask_RCNN-master/samples/shapes/train_shapes.py", line 258, in
layers='heads')
File "D:\py\Mask_RCNN-master\mrcnn\model.py", line 2352, in train
use_multiprocessing=True,
File "D:\anaconda\envs\tensorflow\lib\site-packages\keras\legacy\interfaces.py", line 87, in wrapper
return func(*args, **kwargs)
File "D:\anaconda\envs\tensorflow\lib\site-packages\keras\engine\training.py", line 2011, in fit_generator
generator_output = next(output_generator)
File "D:\anaconda\envs\tensorflow\lib\site-packages\keras\utils\data_utils.py", line 644, in get
time.sleep(self.wait_time)
有大佬知道怎么解决吗,或者有谁出现了相同的问题吗??