Read a large PyTorch model one block at a time

A graph with thousands of nodes is hard to read when every operation is open at once. Start with a smaller question: what does this block take in, and what does it pass to the next block? This walkthr