Closed yihua7 closed 1 year ago
Hi, thanks for the excellent work! I just found that in line here, changing enc = enc * self.mask to enc = enc * self.mask + enc.detach() * (1 - self.mask) is more reasonable and should bring about better quality.
enc = enc * self.mask
enc = enc * self.mask + enc.detach() * (1 - self.mask)
Hi, thanks for the excellent work! I just found that in line here, changing
enc = enc * self.masktoenc = enc * self.mask + enc.detach() * (1 - self.mask)is more reasonable and should bring about better quality.