如何使用legacy / train.py在管道配置中为对象检测api设置keep_checkpoint_max

时间:2019-01-04 22:25:10

标签: python tensorflow object-detection-api

我正在尝试通过将以下行添加到pipeline.config文件中的train_config部分来为我的模型设置以下配置。

  keep_checkpoint_every_n_hours: 1
  keep_checkpoint_max: 100
  save_summaries_secs: 300

但是出现以下错误:

google.protobuf.text_format.ParseError: 109:3 : Message type "object_detection.protos.TrainConfig" has no field named "keep_checkpoint_max".

我已经研究了proto/train_pb2.py中的源代码,实际上,只有keep_checkpoint_every_n_hours在FieldDescriptor中。那么,如果TrainConfig字段中未包含其他两个配置,该如何设置呢?

我正在使用legacy / train.py文件来训练我的模型。

我已经阅读了以下文章,但不确定如何修改tf.estimator的run_config,因为看来老式培训师没有使用此模块。

How to control amount of checkpoint kept by tensorflow estimator?

我看到了以下相关功能要求。无论如何,是否可以围绕当前限制设置这些配置?

https://github.com/tensorflow/models/issues/5246

0 个答案:

没有答案