Output model .pb 

Is there a way to output the .pb file while training the model or if I use the freeze graph method what is the name of the **input node** and **output node**? May you please help me with this. I have tried to freeze the model but I do not know the name of the node I need to output which will be needed for inference. 

Thank you. 