hey! thanks for responding. I thought so too, but it seems like the gradients aren’t flowing backwards. When i check the model.parameters(), i see all the layers that are present. is there a way to retrieve the weight matrices from each layer and observe them as they are trained?