Open cqray1990 opened 3 years ago
offset = conv_sum.matmul(activation.reshape(-1, 1)).reshape(-1) if next_idx in CBL_idx: next_bn = pruned_model.module_list[next_idx][1] next_bn.runningmean.data.sub(offset) else: nextconv.bias.data.add(offset)
offset = conv_sum.matmul(activation.reshape(-1, 1)).reshape(-1) if next_idx in CBL_idx: next_bn = pruned_model.module_list[next_idx][1] next_bn.runningmean.data.sub(offset) else: nextconv.bias.data.add(offset)