Tensorflow verwendet Flags, um Befehlszeilenparameter zu definieren

In diesem Artikel wird hauptsächlich vorgestellt, wie Tensorflow Flags zum Definieren von Befehlszeilenparametern verwendet. Jetzt teile ich ihn mit Ihnen und gebe ihn als Referenz. Schauen wir uns das gemeinsam an

tf definiert tf.app.flags, das zur Unterstützung der Annahme von Befehlszeilenparametern verwendet wird, was der Annahme von argv entspricht.

import tensorflow as tf

#第一个是参数名称，第二个参数是默认值，第三个是参数描述
tf.app.flags.DEFINE_string('str_name', 'def_v_1',"descrip1")
tf.app.flags.DEFINE_integer('int_name', 10,"descript2")
tf.app.flags.DEFINE_boolean('bool_name', False, "descript3")

FLAGS = tf.app.flags.FLAGS

#必须带参数，否则：'TypeError: main() takes no arguments (1 given)';  main的参数名随意定义，无要求
def main(_): 
  print(FLAGS.str_name)
  print(FLAGS.int_name)
  print(FLAGS.bool_name)

if __name__ == '__main__':
  tf.app.run() #执行main函数

Ausführung:

[root@AliHPC-G41-211 test]# python tt.py
def_v_1
10
False
[root@AliHPC-G41-211 test]# python tt.py --str_name test_str --int_name 99 --bool_name True
test_str
99
Richtig

Verwandte Empfehlungen:

So exportieren Sie das Modellnetzwerk von TensorFlow als einzelne Datei

Das obige ist der detaillierte Inhalt vonTensorflow verwendet Flags, um Befehlszeilenparameter zu definieren. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!