cluster center stay the same during training

[FabianGroeger96]

Hi there

During training using the GhostVLAD layer, I have experienced that the cluster centres do not change their value, which means that they always stay at the same place during training. Is this an expected behaviour, or is this a bug in my training process?

To give you some context, I used the GhostVLAD layer after a ResNet18 architecture for an audio task. I adapted your exact code to fit my structure. All the initialization, layers and so on are like in your code.

I would appreciate your help 😄.