get_collection doesn’t work unless I use a checkpoint, what’s the equivalent for a .pb?

Hey guys, so I also made this stackoverflow post if you want to
see it:

So when testing my predictions are correct on the model,
normally I write get_collection(“logits”).

However, for a .pb/inference model I can’t do this anymore
because the saver.restore only works for checkpoints, not .pb, so
even though I can get tensors, I’m not sure how to get this.

What is the equivalent action for the inference model?

submitted by /u/Vendredi46

[visit reddit]


Leave a Reply

Your email address will not be published. Required fields are marked *