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:
my objective is to average the weights and bias of them and put the result on a third model:
How can i achieve that?