I would suggest to add **kwargs here and here as otherwise it does not allow to change some of the parameters, which are not stated in the function while they are in the further constructors (I was trying to change the activation function, thats how I ended up with it)
Hope this would help!