Modifying the agent attention to accept NHWC tensor

LeapLabTHU / Agent-Attention

Official repository of Agent Attention (ECCV2024)

473 stars 35 forks source link

Modifying the agent attention to accept NHWC tensor #40

Open alvinwong64 opened 1 month ago

alvinwong64 commented 1 month ago

Hi i would like to plug in the agentattention module to test for the performance of it on a yolo-segmentation model, how could i modify it to accept input in the form of NHWC, thanks!