I have been trying to implement model ensemble in plt, but haven’t got any elegant solution. The problem is simple: train a ResNet-50 on the same dataset several times with different random seed, when do the inference on a new dataset, I want to average the prediction from all those models. I can certainly do the inference multiple times, each time with a different model and save the prediction in a csv. But, I feel there should be a more elegant solution. Does anyone have any suggestion?