Hi folks, I am trying to build an ensemble with the following code: class PytorchMnistEnsModel(nn.Module): """ Basic MNIST model from github https://github.com/rickiepark/pytorch-examples/blob/master/mnist.ipynb """ def __init__(self, num_models=5): super(PytorchMnistEnsMod…

Here is the modified code. Hopefully this helps someone in the future :slight_smile: class PytorchMnistEnsModel(nn.Module): """ Basic MNIST model from github https://github.com/rickiepark/pytorch-examples/blob/master/mnist.ipynb """ def __init__(self, num_models=5): super(Py…

ValueError:optimizer got an empty parameter list

InnovArul (Arul) October 12, 2018, 3:44pm 2

You could either use nn.ModuleList or nn.Sequential to store the layers. Python lists and dicts aren’t the right ones for storing torch modules.

1 Like