TSC¶

Module Contents¶

class TSC(temperature=0.07, weight_contrast=None, use_aug=False)¶

Bases: torch.nn.Module

use_aug = False¶

temperature = 0.07¶

weight_contrast = None¶

forward(features, quant_idx=None, weight_same_quant=None)¶