Open
Description
在反馈函数中,似乎不是均值呢?而是不断在替换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)
);
因为我看到您,已经写到要计算均值的,可是为什么又注释掉呢?求解~
Metadata
Metadata
Assignees
Labels
No labels