Fix MetricCollection
compatibility with torch.jit.script
#3927
Loading
MetricCollection
compatibility with torch.jit.script
#3927