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.
lucidrains
commented
8 months ago
40
face_maskafter performingmasked_fillonrecon_faces.face_maskwas not properly restored after applying a mask to the reconstructed facial data (recon_faces). This could affect subsequent processes.