Instantly share code, notes, and snippets.

View tensorflow-model-profiling.txt
run_meta = tf.RunMetadata()
with tf.Session(config=config) as sess:
opts = tf.profiler.ProfileOptionBuilder.float_operation()
flops = tf.profiler.profile(sess.graph, run_meta=run_meta, cmd='op', options=opts)
opts = tf.profiler.ProfileOptionBuilder.trainable_variables_parameter()
params = tf.profiler.profile(sess.graph, run_meta=run_meta, cmd='op', options=opts)
print("ops {:,} --- params {:,}".format(flops.total_float_ops, params.total_parameters))
View python-argparse-example.txt
import argparse
import os
parser = argparse.ArgumentParser(description='Generate a saved model.')
parser.add_argument('--export_model_dir', type=str, default='./saved_model/dense-lite', help='export model directory')
parser.add_argument('--model_version', type=int, default=1, help='model version')
parser.add_argument('--model', type=str, default='dense.pb', help='model pb file')
parser.add_argument("--input_tensor", default="image:0", help="input tensor", type=str)
parser.add_argument("--output_tensor", default="Openpose/concat_stage7:0", help="output tensor", type=str)
args = parser.parse_args()
View mac-os-build-misc.txt
use "-undefined dynamic_lookup" instead of "-lpython"
use "-Xpreprocessor -fopenmp -lomp" instead of "-fopenmp"
flags to be added for lpython: "-L/usr/local/opt/libomp/lib -I/usr/include/python2.7" ?
get numpy include dir: np.get_include()