Model Graph Visualisation (KIE)

[amitbcp]

Thanks for open-sourcing the project.
I was trying to visualize the network graph for models (SDMGR) specifically. But since the data input to the model are wrapped in DataContainer from mmcv I was not able.

Do you have any suggestion to create Model's Graph to better understand input and output flows. One Example I was using was : https://github.com/waleedka/hiddenlayer to get the flow& structure of model

[cuhk-hbsun]

@amitbcp Thanks for your question. We will give an end-to-end demo for KIE such that the input data is straightforward and no DataContainer warpped.

[amitbcp]

Thank you @cuhk-hbsun !
Looking forward to it , in order to understand and experiment more around it

[amitbcp]

@cuhk-hbsun , I was revisiting the paper but was not able to find the number of parameters in the model. And since pytorch model summary is not working, can you please with it?