Arshjot Singh Khehra

Results 4 comments of 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