KaiyangZhou/pytorch-center-loss

Some questions about the center loss

taylover-pei opened this issue · 1 comments

Congratuations! You have done a great work. But I have a question about the center loss.
Can you explain why you wrote there two lines?

for param in criterion_cent.parameters():

Thank you very much! Looking forward to your reply!

What I understood is that the author has said the weight parameter will be used to compute the total loss. But when the updation of the centers is being done it is being done without the weight parameter.
Basically, the updates for the centers in this code is nothing but the backpropagation of the mean square error loss.