Solved by albanD in post #4 
              
              
                You should replace the following: 
(you should not give the same Tensor to different parameters. otherwise they have the same memory and so will always have the same value. And if you do an update on one then the other, then you do 2x the update) 
        self.intra_centers = nn.Parameter(self.cente…