Open tanghy2016 opened 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) ); 因为我看到您,已经写到要计算均值的,可是为什么又注释掉呢?求解~
在反馈函数中,似乎不是均值呢?而是不断在替换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])]),(top0_map_Dtype[i])*channels)
);
因为我看到您,已经写到要计算均值的,可是为什么又注释掉呢?求解~
top[0]->cpu_diff() + (i * channels), Dtype(1.0), bout + (static_cast