计算模型参数数与浮点运算量
2017-07-21 本文已影响0人
zhenggeaza
参数数量：
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)
