tensorflow utilise des indicateurs pour définir les paramètres de ligne de commande

Cet article présente principalement comment tensorflow utilise les indicateurs pour définir les paramètres de ligne de commande. Maintenant, je le partage avec vous et le donne comme référence. Jetons un coup d'œil ensemble

tf définit tf.app.flags, qui est utilisé pour prendre en charge l'acceptation des paramètres de ligne de commande, ce qui équivaut à accepter argv.

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函数

Exécution :

[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
Vrai

Recommandations associées :

Comment exporter le réseau de modèles TensorFlow sous forme de fichier unique

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!