更快的rcnn量化意识训练中的例外

时间:2018-11-14 07:23:02

标签: tensorflow object-detection object-detection-api

我得到

File "/home/ubuntu/anaconda3/envs/tensorflow/lib/python3.6/site-packages/tensorflow/contrib/quantize/python/fold_batch_norms.py", line 338, in _ComputeBatchNormCorrections
    match.moving_variance_tensor + match.batch_epsilon)
TypeError: unsupported operand type(s) for +: 'NoneType' and 'float'

enter image description here 使用tensorflow对象检测api在更快的rcnn架构上进行量化感知训练时发生异常。 我能够成功地将其用于ssd mobilenet体系结构。我在GPU和对象检测API的主分支上使用TF 1.11.0。 我添加了

graph_rewriter {
  quantization {
    delay: 48
    weight_bits: 8
    activation_bits: 8
  }
}

到我的pipeline.config文件进行量化。我尝试设置--fine_tune_batch_norm = true,但没有帮助。

0 个答案:

没有答案