从文件加载 TensorFlow 标志(TF 版本 > 1.4)



我想将FLAGS(tf.app.flags.FLAGS(存储在一个文件中,稍后再重新加载。在TensorFlow 1.4之前,我使用以下代码重新加载:

with open(config_file, 'r') as f:
config = json.load(f)
FLAGS.__flags.update(config)

从TensorFlow 1.5开始,FLAGS的底层实现发生了变化。有人知道用存储在文件中的值更新FLAGS的方法吗?

尝试:

with open(config_file, 'r') as f:
config = json.load(f)
for name, value in config.items():
FLAGS.__flags[name].value = value

最新更新