最后一步中的Tensorflow训练错误

Question

我正在尝试训练我自己的张量流对象检测模型，我遵循了this和this教程，在最后一步中我尝试运行此命令

var = int(input("Enter: "))
if var == 1

但我收到此错误

> python train.py --logtostderr --train_dir=training/ --
        pipeline_config_path=training/ssd_mobilenet_v1_pets.config

我该如何解决？

Answer 1

我通过更改.config文件解决了这个问题。我使用ssd_mobilenet_v1_coco.config而不是ssd_mobilenet_v1_pets.config