Summary problem about inputsize

username: Gorgen
cooked: <p>hello, I am new to pytorch.</p>
<p>import torch<br>
import torch.nn as nn<br>
import torch.nn.functional as F<br>
from torchsummary import summary<br>
summary(model, input_size = input_size)<br>
input_size = [ (14,1,1024), (14,1,1024), (14,1,1024), (14,1,1024),(14,1,384),(14,1,768), (14,1,768), (14,1,768), (14,1,768), (14,1,768), (14,1,9), (1,14)]</p>
<p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/6/e/6e760435113a593c171eb346e8da7ba44f8a4a93.png" data-download-href="https://discuss.pytorch.org/uploads/default/6e760435113a593c171eb346e8da7ba44f8a4a93" title="image"><img src="https://discuss.pytorch.org/uploads/default/original/3X/6/e/6e760435113a593c171eb346e8da7ba44f8a4a93.png" alt="image" data-base62-sha1="fLbmZkL8WbXoVV10bmnjhH7So4r" width="690" height="179" data-dominant-color="F8F8FA"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">972×253 11.6 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div><br>
However, model forward need r1, r2, r3, r4, x1, x2, x3, o1, o2, qmask, umask, att2, return_hidden parameter. it will exceed two. How to fix this problem? Thanks, best wishes</p>

username: Gorgen
cooked: <p>sorry. I forget to attch the dataloader defination. Thanks best wishes</p>
<p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/a/f/afd8b5a30f3d13b87b16f7ccf507f6a6cec169aa.png" data-download-href="https://discuss.pytorch.org/uploads/default/afd8b5a30f3d13b87b16f7ccf507f6a6cec169aa" title="BV(4AA$)8CU{QZFR7S{{O12"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/a/f/afd8b5a30f3d13b87b16f7ccf507f6a6cec169aa_2_690x222.png" alt="BV(4AA$)8CU{QZFR7S{{O12" data-base62-sha1="p5BR1TFIdLs3NHj9zoWM6gsH0bo" width="690" height="222" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/a/f/afd8b5a30f3d13b87b16f7ccf507f6a6cec169aa_2_690x222.png, https://discuss.pytorch.org/uploads/default/original/3X/a/f/afd8b5a30f3d13b87b16f7ccf507f6a6cec169aa.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/a/f/afd8b5a30f3d13b87b16f7ccf507f6a6cec169aa.png 2x" data-dominant-color="2E2E2E"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">BV(4AA$)8CU{QZFR7S{{O12</span><span class="informations">855×276 16.8 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>
<p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/4/6/46bf02a8d0069412c6ff97583fcb161a1d8ab0cc.png" data-download-href="https://discuss.pytorch.org/uploads/default/46bf02a8d0069412c6ff97583fcb161a1d8ab0cc" title="image"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/4/6/46bf02a8d0069412c6ff97583fcb161a1d8ab0cc_2_690x496.png" alt="image" data-base62-sha1="a5QEY7pJBDLttYAD1xXwt1KGlnS" width="690" height="496" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/4/6/46bf02a8d0069412c6ff97583fcb161a1d8ab0cc_2_690x496.png, https://discuss.pytorch.org/uploads/default/original/3X/4/6/46bf02a8d0069412c6ff97583fcb161a1d8ab0cc.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/4/6/46bf02a8d0069412c6ff97583fcb161a1d8ab0cc.png 2x" data-dominant-color="2F2F2F"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">936×674 61.6 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>
<p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/f/e/fe6293d35b427672e8418fc58a8aae8ffa08f7a5.png" data-download-href="https://discuss.pytorch.org/uploads/default/fe6293d35b427672e8418fc58a8aae8ffa08f7a5" title="image"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/f/e/fe6293d35b427672e8418fc58a8aae8ffa08f7a5_2_690x339.png" alt="image" data-base62-sha1="AiotG61BVk28gC77xzBlm2fPJo9" width="690" height="339" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/f/e/fe6293d35b427672e8418fc58a8aae8ffa08f7a5_2_690x339.png, https://discuss.pytorch.org/uploads/default/optimized/3X/f/e/fe6293d35b427672e8418fc58a8aae8ffa08f7a5_2_1035x508.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/f/e/fe6293d35b427672e8418fc58a8aae8ffa08f7a5.png 2x" data-dominant-color="2E2E2E"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">1095×538 38.4 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>

username: Gorgen
cooked: <p>here is about dataloader defination. Thanks.</p>
<p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/e/f/ef72cb73ea341915899349b1824857d2fa79f30d.png" data-download-href="https://discuss.pytorch.org/uploads/default/ef72cb73ea341915899349b1824857d2fa79f30d" title="image"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/e/f/ef72cb73ea341915899349b1824857d2fa79f30d_2_454x499.png" alt="image" data-base62-sha1="yag3Qr1sC597i4taepqlXmtREEl" width="454" height="499" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/e/f/ef72cb73ea341915899349b1824857d2fa79f30d_2_454x499.png, https://discuss.pytorch.org/uploads/default/optimized/3X/e/f/ef72cb73ea341915899349b1824857d2fa79f30d_2_681x748.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/e/f/ef72cb73ea341915899349b1824857d2fa79f30d.png 2x" data-dominant-color="FAFAFA"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">819×901 38.4 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>

username: Gorgen
cooked: <p>Fixed. The original summary is not suitable for toom many inputs<br>
Here are some materials about it<br>
tensorboard <a href="https://blog.csdn.net/leonardohaig/article/details/8992368" rel="noopener nofollow ugc">https://blog.csdn.net/leonardohaig/article/details/8992368</a><br>
<a href="https://pypi.org/project/pytorch-model-summary/" class="inline-onebox" rel="noopener nofollow ugc">pytorch-model-summary · PyPI</a></p>