Skip to content

Instantly share code, notes, and snippets.

@NTU-P04922004

NTU-P04922004/tf_slim_lab.py

Created November 24, 2019 13:19
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save NTU-P04922004/37c3ddafda69582ccf011c2fc26122fd to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save NTU-P04922004/37c3ddafda69582ccf011c2fc26122fd to your computer and use it in GitHub Desktop.
Download ZIP
Analyze TF model with tf-slim
Raw
tf_slim_lab.py
import tensorflow.contrib.slim as slim
model_vars = tf.trainable_variables()
slim.model_analyzer.analyze_vars(model_vars, print_info=True)
slim.model_analyzer.analyze_ops(tf.compat.v1.get_default_graph(), print_info=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment