tensorflow - 如何在不弃用的情况下修复错误 tensorflow has not attribute 'ConfigProto'？

Question

我正在尝试使用此命令行在我的数据集上训练 Retinanet：

  retinanet-train --batch-size 4 --steps 349 --epochs 50 --weights logos/resnet50_coco_best_v2.1.0.h5 --snapshot-path logos/snapshots csv logos/retinanet_train.csv logos/retinanet_classes.csv

我得到这个错误：

    AttributeError: module 'tensorflow' has no attribute 'ConfigProto'

我知道，这与 Tensorlow 的版本有关，在新版本中 ConfigProto 消失了，但我想在不“重新安装”旧版本1.14的情况下修复它，否则会一团糟。

任何建议将不胜感激，谢谢。

Answer 1

由于tf.ConfigProto在 TF 2.0 中已弃用，tf.compat.v1.ConfigProto()因此通过替换代码tf.ConfigProto()中的出现来代替使用retinanet-train（假设这tf.ConfigProto()是被调用的地方）。此处链接到tensorflow 文档。