ma-xu / Context-Cluster

[ICLR 2023 Oral] Image as Set of Points

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Dimensional problem during the Validation.

sky123CT opened this issue · comments

Hi, Thank you very much for your contribution. Your works are very attractive.

But during the validation of srgementation. There are some Dimensional problem when I try to load the model checkpoint you provide, namely Coc_small_49. Could you please provide correct model checkpoint and backbone checkpoint?
Snipaste_2023-05-14_15-21-09

commented

@sky123CT Thanks for your interest.

The reson is that the released code change conv1x1 to fc for faster inference. See here: #2
We will update this by the end of May of early June since I'm busy recently.

commented

@sky123CT Thanks for your patience. The updated checkpoints are updated, with slightly improvements. Mainly two updates: 1) fix a bug mask.sum to sim.sum; 2) conv1x1 to Linear implementation.

Thank you so much for your work!