版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
在调试DeRPN的时候,训练一段时间就会挂掉。想看看是哪张图引起的崩溃,就想在取minibatch的时候,输出训练的图片名。
方案1:使用print
print "Processing.... ", roidb[i]['image']
方案2:使用logging库
logging.warning(roidb[i]['image'])
方案1的缺点是:输出和GPU训练输出对不上,时常错乱;
方案2可以解决这个问题:
参考链接: