Could you provide codes for calculating FID, Time (s/im), and Memory (GB/im)?

LeapLabTHU / Agent-Attention

Official repository of Agent Attention (ECCV2024)

473 stars 35 forks source link

Could you provide codes for calculating FID, Time (s/im), and Memory (GB/im)? #22

Closed Doraemonzm closed 7 months ago

Doraemonzm commented 7 months ago

Thanks for your excellent work. I noticed that AgentSD performs better than ToMeSD and Stable Diffusion v1.5. Could you provide codes for calculating FID, Time (s/im), and Memory (GB/im)?

tian-qing001 commented 7 months ago

Hi @Doraemonzm, thank you for your attention to our work. As described in section B.2, we calculate FID scores employing pytorch-fid. To assess speed, we calculate the average generation time of all 2,000 samples on a single RTX4090 GPU. GB/img is measured as the total memory usage change when increasing batch size by 1.