sru icon indicating copy to clipboard operation
sru copied to clipboard

Published 20 hours ago • verified publisher icon asappresearch

batch_first flag

Open nankeqin opened this issue 4 years ago • 1 comments

Can I add a batch_first flag? And how?

nankeqin avatar May 12 '21 13:05 nankeqin

hi @nankeqin, there is no batch_first option in SRU forward. You can add a wrapper over SRU module with this option, or transpose your model input into length_first before calling SRU.

taoleicn avatar May 15 '21 01:05 taoleicn