I am having trouble on how to access the weights. I want to add noise to the weights during the training process.