Fixing AttentionModule
test fails after last commit
#738
Loading
AttentionModule
test fails after last commit
#738