[深度学习]tensorflow中flags的使用(代码)

xiaoxiao2025-09-18  5

import tensorflow as tf #第一个是参数名称,第二个参数是默认值,第三个是参数描述 tf.app.flags.DEFINE_string('str_name', 'wuhcao', 'example1') tf.app.flags.DEFINE_integer('int_name', 100, 'example2') tf.app.flags.DEFINE_boolean('bool_name', True, 'example2') 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函数

在文件所在的路径运行代码: 结果为: 在终端修改参数: 结果如下:

转载请注明原文地址: https://www.6miu.com/read-5036552.html

最新回复(0)