openai / glow

Code for reproducing results in "Glow: Generative Flow with Invertible 1x1 Convolutions"
https://arxiv.org/abs/1807.03039
MIT License
3.11k stars 515 forks source link

Training Output Nodes #84

Open jspang16 opened 5 years ago

jspang16 commented 5 years ago

In order to perform inference on the trained model the input and output nodes for encode/decode must be known. Does anyone know how to figure this out / what the naming scheme is for a generic model with given hyperparameters? I've tried visualizing my trained model in tensorflow but the graph is extremely cluttered compared to that of the pretrained model making it impossible to decipher outputs / input nodes. Any help would be appreciated.