update

pixel_generator/vec2face/model_vec2face.py

@@ -303,7 +303,7 @@ class MaskedGenerativeEncoderViT(nn.Module):
                 id_loss = torch.mean(1 - torch.cosine_similarity(out_feature, rep))
             else:
                 distance = 1 - torch.cosine_similarity(out_feature, class_rep)
-                id_loss = torch.mean(torch.where(distance > 0.3, distance, torch.zeros_like(distance)))
+                id_loss = torch.mean(torch.where(distance > 0.1, distance, torch.zeros_like(distance)))
             quality = quality_model(image)
             norm = torch.norm(quality, 2, 1, True)
             q_loss = torch.where(norm < q_target, q_target - norm, torch.zeros_like(norm))