cyto_dl.nn.losses.vic_reg module#

class cyto_dl.nn.losses.vic_reg.VICRegLoss(num_features, sim_coeff=5, std_coeff=5, cov_coeff=1)[source]#

Bases: Module

forward(x, y)[source]#
cyto_dl.nn.losses.vic_reg.off_diagonal(x)[source]#