Hi,
I followed the instructions and trained 7 model and adapted your code for me. But when I pass an input to the model , I get following error:
File "ModelEnsemble.py", line 162, in forward
    x = self.classifier(F.relu(x))
  File "/home/user/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/user/.local/lib/python3.6/site-packages/torch/nn/modules/linear.py", line 92, in forward
    return F.linear(input, self.weight, self.bias)
  File "/home/user/.local/lib/python3.6/site-packages/torch/nn/functional.py", line 1406, in linear
    ret = torch.addmm(bias, input, weight.t())
RuntimeError: size mismatch, m1: [1 x 1053696], m2: [11648 x 7] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:268

All my trained models are based on densenet169, So my in_features= 1664.