Using a random dataset, your weights seem to upgrade fine:
train_loader = DataLoader(
    torch.utils.data.TensorDataset(
        torch.randn(10, 3, 64, 64),
        torch.randint(0, 2, (10,))
    )
)