iTPN icon indicating copy to clipboard operation
iTPN copied to clipboard

fpn_dim in itpn_base_pixel checkpoint

Open VGrondin opened this issue 1 year ago • 1 comments

Hello, thanks for your work and congrats for the ICRA2023 acceptation, it was really interesting to read. I am experimenting with iTPN for self-supervised learning, but the checkpoint provided (itpn_base_pixel.pth on Drive) seems to have fpn_dim=512. The default config should be 256 from what I see in the code. Is it possible to upload the fpn_dim=256 version? It would help for the downstream task since RPN takes 256 as inputs.

Have a great day.

VGrondin avatar Mar 22 '23 14:03 VGrondin

I will update the checkpoint with fpn dim=256. Thanks

---- Replied Message ---- | From | Vincent @.> | | Date | 03/22/2023 22:06 | | To | @.> | | Cc | @.***> | | Subject | [sunsmarterjie/iTPN] fpn_dim in itpn_base_pixel checkpoint (Issue #3) |

Hello, thanks for your work it was really interesting to read. I am experimenting with iTPN for self-supervised learning, but the checkpoint provided (itpn_base_pixel.pth on Drive) seems to have fpn_dim=512. The default config should be 256 from what I see in the code. Is it possible to upload the fpn_dim=256 version? It would help for the downstream task since RPN takes 256 as inputs.

Have a great day.

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>

sunsmarterjie avatar Mar 22 '23 14:03 sunsmarterjie