ma-xu / Context-Cluster

Hi @ma-xu ,

I want to use Context-Cluster to do some down-stream tasks. However, I find that I can not load you provide pre-trained model. This may caused by the MLP layer

Context-Cluster/models/context_cluster.py

Line 188 in c307ab1

class Mlp(nn.Module):

.

When will you upload the new pre-trained model?

@dream-toy Thanks for your interest. We will release new checkpoint by the end of this month due to limited computational resources.

ok, Thank you.

You can modify a few variable names in the Cluster class of context_cluster.py. self.f -> self.fc1 self.proj -> self.fc2 self.v -> self.fc_v

@dream-toy @DengFlagellants

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.

Closed no further discussions. Feel free to reopen it if any further questions.

Can not load the pre-trained model.