python - 在不使用命令行的情况下训练 Tensorflow 模型

Question

我想在不产生新进程的情况下从另一个 Python 脚本调用这个main(_)函数（这样更容易调试）。但是，该函数是为使用命令行参数而编写的。直接从另一个函数调用该函数的最干净的方法是什么？

Answer 1

您可以导入FLAGS然后定义所需的参数（train_data、eval_data、save_path）。

In [13]: from tensorflow.models.embedding.word2vec_optimized import FLAGS
In [14]: from tensorflow.models.embedding.word2vec_optimized import main
In [16]: main(_)
--train_data --eval_data and --save_path must be specified.
An exception has occurred, use %tb to see the full traceback.
In [17]: FLAGS.train_data = "this"
In [18]: FLAGS.eval_data = "that"
In [19]: FLAGS.save_path = "some_path"
In [20]: main(_)
I tensorflow/core/common_runtime/local_device.cc:40] Local device intra op parallelism threads: 8