Idk

Results 2 issues of Idk

Why do mappings of attention maps and hidden states are different, can you explain a little bit? Also when I want to plug-in my own model second mapping return "index...

There's domain-invariant perceptual loss in their paper and implementation. But I don't see it here, I think it's space for improvement. Also, I might add this functionality if you want.