1. 程式人生 > >Tensorflow執行

Tensorflow執行

1.FLAGS = tf.app.flags.FLAGS 執行main函式之前首先進行flags的解析,也就是說TensorFlow通過設定flags來傳遞tf.app.run()所需要的引數,我們可以直接在程式執行前初始化flags,也可以在執行程式的時候設定命令列引數來達到傳參的目的。 flags = tf.app.flags FLAGS = flags.FLAGS flags.DEFINE_float('learning_rate', 0.01, 'Initial learning rate.') flags.DEFINE_integer('epoch_number', None, 'Number of epochs to run trainer.') flags.DEFINE_integer("thread_number", 10 , "Number of thread to read data") flags.DEFINE_string("mode", "train", "Option mode: train, train_from_scratch, inference")