Untraced function error

[Jenny_Chen1]

Hi!

My models are passing checks and the calculations for mae are running for both the training and the testing. However, whenever I try to save my model. I get a untraced function error - like this

Could someone point me in the right direction?

Thanks!

[balaji.ambresh]

I’ve asked the staff for a response on using saved model format instead of .h5 file to save model weights. For now, please ignore the warnings.

That said, your model has a shape mismatch issue. The model output shape doesn’t match the expected shape.