code: def test_momentum_works(self):
      model = nn.Sequential(
          nn.Linear(32, 32),
          nn.ReLU(),
      )
      model_momentum = copy.deepcopy(model)
      update_momentum(model, model_momentum, 0.99)
  
imports: import copy
  import unittest
  import torch
  import torch.nn
  from lightly.models import utils
  from lightly.models.utils import _no_grad_trunc_normal, activate_requires_grad, batch_shuffle, batch_unshuffle, deactivate_requires_grad, nearest_neighbors, normalize_weight, update_momentum
label_desc: Use image node_id 4 for calling the TestModelUtils obj's underlying member method code with example usage: obj.test_momentum_works() without return types