guoyilin / caffe

some new implementation of caffe
Other
24 stars 8 forks source link

some error in TripletSamplingLayer? #4

Open tanghy2016 opened 8 years ago

tanghy2016 commented 8 years ago

在反馈函数中,似乎不是均值呢?而是不断在替换bottom里面的值,最终保留的是最后一次的结果吧? 是不是可以考虑,先caffe_set(bottom[0]->count(), 0, bottom[0]->cpu_diff()); 后面再调用caffe_cpu_axpby的时候,改写为: caffe_cpu_axpby( channels, Dtype(1.0 / image_count_dtype[static_cast(top0_map_Dtype[i])]),
top[0]->cpu_diff() + (i * channels), Dtype(1.0), bout + (static_cast(top0_map_Dtype[i])*channels) ); 因为我看到您,已经写到要计算均值的,可是为什么又注释掉呢?求解~