Anyone have any good ways to grab just the weights and not the biases from a module?
I can think of the following kinda hacky solution to filter by dimension since all biases should be 1 dim, maybe? :
for param in model.parameters():
    if param.dim() > 1:
        #this is a weight parameter