@ptrblck thanks for the info. I am still new to pytorch, so I don’t really know the package ecosystem yet. I will check out torchviz. That is great. Right now I am just putting the model code into pdb and then checking the sizes of the output tensors. That seems to work well–better than tensorflow, anyway  But it would be nice to have some visual way to visually inspect the model to make sure I did not miss anything.
Thanks again.