Fix (nn): state_dict load for unpacked in_proj in MHA (#654) #1276
| Job | Run time |
|---|---|
| 4m 18s | |
| 2m 25s | |
| 4m 4s | |
| 4m 4s | |
| 3m 9s | |
| 3m 48s | |
| 5m 1s | |
| 2m 44s | |
| 4m 27s | |
| 3m 23s | |
| 3m 17s | |
| 4m 3s | |
| 4m 51s | |
| 2m 41s | |
| 4m 21s | |
| 3m 56s | |
| 2m 11s | |
| 4m 28s | |
| 2m 51s | |
| 2m 1s | |
| 3m 10s | |
| 3m 42s | |
| 1m 56s | |
| 6m 35s | |
| 2m 45s | |
| 2m 17s | |
| 3m 13s | |
| 2m 30s | |
| 2m 16s | |
| 3m 12s | |
| 1h 43m 39s |