Arshjot Singh Khehra
Arshjot Singh Khehra
We can set the input shape while defining the model as shown below: ``` class CustomModel(tf.keras.Model): def __init__(self, num_classes=10): super(CustomModel, self).__init__(name='custom_model') self.num_classes = num_classes self.flatten = tf.keras.layers.Flatten() self.dense_1 = tf.keras.layers.Dense(128,...
Hey @murarugeorgec, did you make any progress? Otherwise, I can take up this issue.
Thank you guys! I've just submitted a PR to fix this :)
I had this issue too, try using the full path (absolute path) of the checkpoint instead of the relative path