Как я могу исправить ошибку, у тензорного потока нет атрибута 'ConfigProto' без устаревания? - PullRequest
0 голосов
/ 09 января 2020

Я пытаюсь обучить Retin anet в моем наборе данных с помощью этой командной строки:

  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 , потому что иначе это будет беспорядок.

Любое предложение будет высоко ценится, спасибо.

1 Ответ

0 голосов
/ 09 января 2020

Поскольку tf.ConfigProto устарело в TF 2.0, вместо этого используйте tf.compat.v1.ConfigProto(), заменив вхождения tf.ConfigProto() в коде retinanet-train (при условии, что именно здесь вызывается tf.ConfigProto()). Ссылка на tenorflow do c здесь.

...