What's the output node name for the saved checkpoint?

yuanming-hu / exposure

Learning infinite-resolution image processing with GAN and RL from unpaired image datasets, using a differentiable photo editing model.

MIT License

767 stars 156 forks source link

What's the output node name for the saved checkpoint? #33

Open dreamibor opened 6 years ago

dreamibor commented 6 years ago

Hi,

What's the output node name for the checkpoint file? I am trying to freeze the model as a protobuf file.

Thanks

yuanming-hu commented 6 years ago

I'm not sure if I understand your question. If you are looking for the code to save the model, it is here: https://github.com/yuanming-hu/exposure/blob/master/net.py#L381

dreamibor commented 6 years ago

Hi, thank you for your reply. I want to freeze the model from the TensorFlow checkpoint to .pb file. I am wondering which node is used for output the image or where are the output nodes in the network if I only want to do inference?