텐서플로우는 플래그를 사용하여 명령줄 매개변수를 정의합니다.

이 글에서는 주로 텐서플로우에서 명령줄 매개변수를 정의하기 위해 플래그를 사용하는 방법을 소개합니다. 이제 이를 공유하고 참고용으로 제공합니다. 함께 살펴보겠습니다

tf는 argv를 허용하는 것과 동일한 명령줄 매개변수 허용을 지원하도록 tf.app.flags를 정의합니다.

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

실행:

[root@AliHPC-G41-211 테스트]# python tt.py
def_v_1
10
False
[root@AliHPC-G41-211 테스트]# python tt.py - -Str_name test_str --int_name 99 ---bool_name true test_str
99
true

related 권장 사항 :

TensorFlow의 모델 네트워크를 단일 파일로 내보내는 방법

위 내용은 텐서플로우는 플래그를 사용하여 명령줄 매개변수를 정의합니다.의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!