ETC¶

Module Contents¶

class ETC(num_times, temperature, weight_neg, weight_pos)¶

Bases: torch.nn.Module

num_times¶

weight_neg¶

weight_pos¶

temperature¶

forward(topic_embeddings)¶

compute_loss(anchor_feature, contrast_feature, temperature, self_contrast=False, only_pos=False, all_neg=False)¶