Tensorflow code is quite the mess. The whole tree structure of your model makes so much more sense in pytorch.,(reply>), When TF engineers made their libraries, they did not expect it to be used by millions of people all around the world one day.