Can't save custom subclassed model

Question

Inspired by tf.keras.Model subclassing I created custom model.
I can train it and get successfull results, but I can\'t save it.
I use python3.6 wit

Answer 1

use model.predict before tf.saved_model.save