def METHOD_NAME(model, model_ref):
    count = 0
    for param, param_ref in zip(model.parameters(), model_ref.parameters()):
        assert (param != param_ref).any(), "param {} with shape {} not updated!! \n{}\n{}".format(
            count, param.shape, param, param_ref
        )
        count += 1