ivus-segmentation / scripts /model_check.py
Aditya2162's picture
Upload folder using huggingface_hub
1d197a4 verified
import tensorflow as tf
m = tf.saved_model.load("./model")
print("Has call:", callable(m))
print("Num variables:", len(getattr(m, "variables", [])))
print("Num trainable:", len(getattr(m, "trainable_variables", [])))
print("Signatures:", list(getattr(m, "signatures", {}).keys()))