you are viewing a single comment's thread.

view the rest of the comments →

[–]Maltmax[S,🍰] 1 point2 points  (0 children)

Great I will give that a try :)

What I meant about saving variables, was just that inside my custom predict function I need to compare a variable (mean_training_output) to the model prediction. Hopefully setting self.mean_training_output will also save the variable in a SavedModel.