suprosanna / relationformer

Apache License 2.0
107 stars 15 forks source link

question about hid_dim when using pretrained deformable DETR #7

Open long8v opened 2 years ago

long8v commented 2 years ago

Hi there, thank you for providing your work as open source. I can understand your work more thoroughly with your repo. I have a curiosity about HIDDEN_DIM https://github.com/suprosanna/relationformer/blob/scene_graph/configs/scene_2d.yaml#L40 I believe you must have used pretrained weight from https://github.com/fundamentalvision/Deformable-DETR, because I cannot find mentioning about pretraining in your paper. However, as far as I know, all deformable DETR pretrained weight is set to have 256 hidden_dim, while your configuration says 512. Was there any trick for loading pretrained weight or any public available deformable DETR model with 512 hidden dim?