osmr / imgclsmob

Sandbox for training deep learning networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to visualize the network and use it in tensorflow 2.0?

atmadeep opened this issue · comments

Might be a really naive question. Relatively new to keras and tensorflow.
The example cited in the gist returns a ops.EagerTensor instance and I was wondering how can I convert it into a tf.keras model instance and use it? (in order to freeze and remove layers)
tf2cv

The summary method returns just the number of parameters as opposed to individual layers.