fix: prob attention shape error while bs>1 (#50) #154
Job | Run time |
---|---|
3m 25s | |
17m 28s | |
16m 15s | |
18m 3s | |
23m 20s | |
24m 54s | |
29m 12s | |
29m 8s | |
29m 20s | |
28m 53s | |
30m 11s | |
4h 10m 9s |
Job | Run time |
---|---|
3m 25s | |
17m 28s | |
16m 15s | |
18m 3s | |
23m 20s | |
24m 54s | |
29m 12s | |
29m 8s | |
29m 20s | |
28m 53s | |
30m 11s | |
4h 10m 9s |