Hello,
I try your code snippet on my machine, it works fine.
[{'nesterov': False, 'weight_decay': 0, 'lr': 0.001, 'dampening': 0, 'params': [Parameter containing:
tensor([[ 0.2869, -0.1661,  0.1697,  0.4048, -0.3160],
        [-0.1766,  0.2777, -0.0006, -0.0400,  0.1492]], requires_grad=True), Parameter containing:
tensor([-0.2891,  0.1473], requires_grad=True)], 'momentum': 0.9}, {'nesterov': False, 'weight_decay': 0, 'lr': 0.001, 'dampening': 0, 'params': [tensor([[ 1.1468, -0.0929,  0.4642,  0.1088, -0.5372],
        [ 1.3683, -0.6719,  0.6465, -0.5709,  0.1006],
        [-0.9102,  0.1764,  0.4354,  1.3183, -0.1849],
        [-0.0164, -0.8461, -0.5933,  0.8613,  2.3292],
        [ 0.6723,  0.7489,  0.4820,  1.0486,  0.4104]])], 'momentum': 0.9}]