How do I plot a Keras/Tensorflow subclassing API model?
I’ve found some workaround to plot with the model sub-classing API. For the obvious reason Sub-Classing API doesn’t support Sequential or Functional API like model.summary() and nice visualization using plot_model. Here, I will demonstrate both. class my_model(keras.Model): def __init__(self, dim): super(my_model, self).__init__() self.Base = keras.keras.applications.VGG16( input_shape=(dim), include_top = False, weights=”imagenet” ) self.GAP = L.GlobalAveragePooling2D() self.BAT … Read more