Function cm

project_utils/cluster_memory_utils.py:35–39

(inputs, indexes, features, momentum=0.5, device=None)

def cm(inputs, indexes, features, momentum=0.5, device=None):
    if device is None:
        return CM.apply(inputs, indexes, features.to(inputs.device), torch.Tensor([momentum]).to(inputs.device))
    else:
        return CM.apply(inputs.to(device), indexes, features.to(device), torch.Tensor([momentum]).to(device))


class CM_Hard(autograd.Function):