update

pixel_generator/vec2face/model_vec2face.py

@@ -318,7 +318,7 @@ class MaskedGenerativeEncoderViT(nn.Module):
                 yaw_loss = torch.abs(pose - torch.abs(pose_info[:, 1].clip(min=-90, max=90)))
                 pose_loss = torch.mean(yaw_loss)
             q_loss = torch.mean(q_loss)
-            if pose_loss > 5 or id_loss > 0.3 or q_loss > 1:
+            if pose_loss > 5 or id_loss > 0.1 or q_loss > 1:
                 i -= 1
             loss = id_loss * 100 + q_loss + pose_loss
             optm.zero_grad()