I could send the parameters during the model training but if you print them you’ll see the initiative parameters are there and not the new one. it seems like once you initiated the layer with certain parameters you can just change normally. There must be another way to do it .