[ASK] How can I save SASRec model for re-training and prediction?

[beomso0]

I have tried to save trained SASRec model.

pickle, tf.saved_model.save, model.save(), and surprise.dump are not working.

While saving, I got warning saying 'Found untraced functions', and while loading, 'AttributeError: 'SASREC' object has no attribute 'seq_max_len''.

Plz someone let me know how to save and load SASRec model!

[RanjanRavi2398]

Hey @beomso0 , If you want save model in TensorFlow. I have shared one link below. that's might be help you. https://cv-tricks.com/tensorflow-tutorial/save-restore-tensorflow-models-quick-complete-tutorial/ please go through this.

[YoungWoong-Cho]

Hey @beomso0 , If you want save model in TensorFlow. I have shared one link below. that's might be help you. https://cv-tricks.com/tensorflow-tutorial/save-restore-tensorflow-models-quick-complete-tutorial/ please go through this.

Blindly applying this solution causes bunch of other errors due to the tf v2 compatibility issue. Creating a Saver is impossible when the eager_execution is turned on; when it is turned off, the training won't happen. It would be greatly appreciated if save function is implemented for SASRec.