Callback error from train_mnist function

[psantillo]

I seem to be getting an attribute error: “myCallback” object has no attribute ‘set_model’ from the below code:

[balaji.ambresh]

Please click my name and send your notebook as an attachment.

[psantillo]

Thank you sending now

[Jaime_Gonzalez]

Did you manage to fix it? I have the same problem

[balaji.ambresh]

Please click my name and send your notebook as an attachment.

[psantillo]

[code reference removed]

Here’s my callback and train_mnist function. My issue was I wasn’t inheriting from the correct class for tf.keras.callback.Callback

[SHAHID_KAMAL]

I am having the same problem . please tell me how to fix it.

[balaji.ambresh]

Does your callback inherit from tf.keras.callbacks.Callback ?