Input to AFP module

[Sparsh913]

Dear Authors,

Excellent work on the foundation level!
As per the architecture shown in the paper, AFP module should be inputted feature maps outputted from the image encoder. However, in the script idisc/models/idisc.py, I notice that AFP module cross attends the output from the pixel decoder. I'm confused. Is there any gap in my understanding? Please let me know

This is the line of code I'm referring to:

idisc/idisc/models/idisc.py

Line 68 in 5d6cea2

idrs = self.afp(decoder_outputs)

Thanks & Regards
Sparsh Garg