Implement model ensemble

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?


hello, not sure if having the very same model just with different weights make such a difference, in theory, your ensembles shall be composed of weak classifiers it various architectures to ver the ground…