Comment by xms17189

How are you generating the architecture graph from model configs, and do you plan to surface tensor shapes or layer-level parameter counts as well?