Closed qixuema closed 8 months ago
hey happy new years! yea i'll get back to this in jan
in the spirit of open source, always welcoming PRs
Thank you for your prompt reply. I hope your holiday season is filled with joy and relaxation.
In the meantime, I'll consider working on a Pull Request to address this issue. Once I have something ready, I'll submit a PR for your review.
Hi Phil,
Happy New Year! 🎉🎉🎉
I encountered a minor issue in the code when I set
return_recon_faces
toTrue
. During the execution of the autoencoder's forward method, I noticed thatface_mask
is rearranged.However, after applying
masked_fill
torecon_faces
, it seems that the shape offace_mask
isn't restored. This is causing an error when trying to perform a repeat operation onface_mask
later in the process.Perhaps the shape of
face_mask
could be restored right after themasked_fill
operation.Best regards, Xueqi Ma