参数数量:
np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()])
浮点运算量:
tf.contrib.tfprof.model_analyzer.print_model_analysis(tf.get_default_graph(), tfprof_options=tf.contrib.tfprof.model_analyzer.FLOAT_OPS_OPTIONS)
参数数量:
np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()])
浮点运算量:
tf.contrib.tfprof.model_analyzer.print_model_analysis(tf.get_default_graph(), tfprof_options=tf.contrib.tfprof.model_analyzer.FLOAT_OPS_OPTIONS)
本文标题:计算模型参数数与浮点运算量
本文链接:https://www.haomeiwen.com/subject/cgnekxtx.html
网友评论