Are you saying it’s as simple as switching the weights in the shapes? So on weights_2_3 instead of shape being (64, 128) it should be (128,64)?
Are you saying it’s as simple as switching the weights in the shapes? So on weights_2_3 instead of shape being (64, 128) it should be (128,64)?