Untraced function error


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

Grading Output

Could someone point me in the right direction?


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.