search

neuroailab / LocalAggregation-Pytorch

88 stars 17 forks source link

A question about computing Ci by k-mean #6

Open MrYaoH opened 3 years ago

MrYaoH commented 3 years ago

Hi, chengxuz, I have a question. In LA part, the loss backpropogates before k-mean updation. Does this mean that the first data batch did not define Ci as mentioned in the paper？

chengxuz commented 3 years ago

At the begining of the training, which is before any loss computation, Ci would be computed using the saved memory bank.