Hi, so i have this network:
class Network(nn.Module):
    def __init__(self):
        super(Network, self).__init__()

        self.conv2d = nn.Sequential(
                nn.Conv2d(1, 64, (3,6), (1, 1)),
                nn.Hardsigmoid()
            )

    def forward(self, input):
        output = self.conv2d(input)
        
        return output

The network is trained on two different dataset so to obtain two different models:
model1
model2

my objective is to average the weights and bias of them and put the result on a third model:
model3
How can i achieve that?