To find out input size to insert summary

1

Hi, today I was running GNMTv2 of PyTorch.

I want to know the total parameter size of GNMT, But I cannot find how to work with pytorch summary. I was looking at this, It needs “input size of GNMT”.

Now I set the option of GNMT, layer=16, sentence size fix to 30.

or I want to know how can i check the total parameter size of the model like transformer.

Thanks.