Fix (nn): state_dict load for unpacked in_proj in MHA (#654) #1238
| Job | Run time |
|---|---|
| 3m 41s | |
| 2m 25s | |
| 2m 52s | |
| 3m 42s | |
| 3m 12s | |
| 2m 37s | |
| 2m 18s | |
| 3m 37s | |
| 3m 28s | |
| 3m 12s | |
| 3m 24s | |
| 2m 11s | |
| 2m 21s | |
| 5m 22s | |
| 3m 23s | |
| 3m 13s | |
| 4m 6s | |
| 2m 18s | |
| 2m 15s | |
| 4m 52s | |
| 3m 59s | |
| 3m 10s | |
| 3m 15s | |
| 1m 52s | |
| 2m 47s | |
| 3m 1s | |
| 3m 35s | |
| 2m 5s | |
| 1m 18s | |
| 2m 35s | |
| 1m 54s | |
| 2m 11s | |
| 1m 22s | |
| 2m 4s | |
| 2m 18s | |
| 2m 18s | |
| 2m 26s | |
| 2m 5s | |
| 1m 50s | |
| 1m 55s | |
| 2m 42s | |
| 2m 49s | |
| 2m 28s | |
| 1m 57s | |
| 1m 38s | |
| 2m 3s | |
| 2m 36s | |
| 2m 57s | |
| 1m 47s | |
| 2m 1s | |
| 1m 38s | |
| 1m 43s | |
| 4m 10s | |
| 2m 55s | |
| 2h 25m 53s |