Thank you for sharing the code, but I couldn't find the code for the Masked MI loss mentioned in the paper. So, I wanted to ask if it's my oversight or if this part of the code hasn't been made public yet.
Hi, the training code (including the proposed masked MI loss) is still reorganized.
Our implementation of the masked MI loss heavily depends on Man Zhou`s implementation.
Thank you for sharing the code, but I couldn't find the code for the Masked MI loss mentioned in the paper. So, I wanted to ask if it's my oversight or if this part of the code hasn't been made public yet.