在运行maskrcnn时,会碰到训练不动的问题,就卡在这儿
UserWarning: Converting sparse IndexedSlices to a dense Tensor of unknown shape. This may consume a large amount of memory.
"Converting sparse IndexedSlices to a dense Tensor of unknown shape. "
Epoch 1/10
image_id 31
然后根据一些博主建议,用最新的mrcnn代码代码试了试,这回会报错了
Failed to get device properties, error code: 30
然后用nvidia-smi在cmd试了试,果然GPU又挂了,之前就一直出现GPU训练一次后会挂掉,必须重启电脑才行
Unable to determine the device handle for GPU 0000:01:00.0: GPU