tanghy2016
tanghy2016
### Describe the bug error log: ``` Traceback (most recent call last): File "/root/project/mmrazor/mmrazor/models/algorithms/autoslim.py", line 69, in _init_pruner pseudo_pruner.prepare_from_supernet(pseudo_architecture) File "/root/project/mmrazor/mmrazor/models/pruners/structure_pruning.py", line 152, in prepare_from_supernet pseudo_img = supernet.forward_dummy(pseudo_img) File "/root/project/mmrazor/mmrazor/models/architectures/base.py",...
在反馈函数中,似乎不是均值呢?而是不断在替换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) ); 因为我看到您,已经写到要计算均值的,可是为什么又注释掉呢?求解~