Co-DETR icon indicating copy to clipboard operation
Co-DETR copied to clipboard

summary of model

Open prsbsvrn opened this issue 1 year ago • 4 comments

Hello, I need detailed information about the model, and I would like to use torchinfo's summary function. Could you advise on where I should incorporate the following code? from torchinfo import summary input_size = (3, 224, 224) # Example input size (C, H, W) for a typical image print("Summary of model") summary(model, input_size)

I have already attempted to integrate it into train.py within both the tools and apis folders of the mmdet directory. Your guidance would be greatly appreciated.

prsbsvrn avatar Apr 27 '24 11:04 prsbsvrn

Some operations (e.g., deformable attention) in the model do not support model summary.

TempleX98 avatar May 03 '24 08:05 TempleX98

Thanks for your response. How can I learn about flops and the impact of each part of the model on flops? Can you guide me?

prsbsvrn avatar May 04 '24 14:05 prsbsvrn

The total FLOPs of Co-DINO-5scale is the same as DINO-5scale since our training technique does not introduce additional inference costs. The FLOPs of the DINO model is presented in Table 1 of its paper here.

TempleX98 avatar May 04 '24 19:05 TempleX98

Thank you, what about Co Deformable DETR?

prsbsvrn avatar May 05 '24 10:05 prsbsvrn