Fix MetricCollection compatibility with torch.jit.script
#9951
| Job | Run time |
|---|---|
| 2m 35s | |
| 3m 44s | |
| 16s | |
| 3s | |
| 25s | |
| 2m 26s | |
| 2m 39s | |
| 1m 50s | |
| 1m 28s | |
| 2m 42s | |
| 2m 39s | |
| 1s | |
| 20m 48s |