Running FedMA with simple cnn

[amosy3]

Hi,
Very good job! I love your method.
I tried to run FedMA using the "simple-cnn" model, but there is a miss-match in the size of matched_cnn (which is bigger than the original model) and the weights after alignment.

The error is:
File "../main.py", line 97, in trans_next_conv_layer_backward
reshaped = layer_weight.reshape(reconstructed_next_layer_shape).transpose(1, 0, 2, 3).reshape(next_layer_shape[0], -1)
ValueError: cannot reshape array of size 3750 into shape (15,25,5,5)

[Anubhav3084]

this error is because they are using the global model size for initializing local models. I tried to change that in my local machine but that created some other errors down the line. Did you solve this issue?