Overview: This commit addresses the issue of restoring the shape of face_mask after performing masked_fill on recon_faces.
Background: An issue was identified where the shape of face_mask was not properly restored after applying a mask to the reconstructed facial data (recon_faces). This could affect subsequent processes.
40
face_mask
after performingmasked_fill
onrecon_faces
.face_mask
was not properly restored after applying a mask to the reconstructed facial data (recon_faces
). This could affect subsequent processes.