timingisnow
timingisnow
### Search before asking - [X] I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and [discussions](https://github.com/ultralytics/yolov5/discussions) and found no similar questions. ### Question as title, in val.py, if I use .pt file...
in get_prune_threshold function, percent threshold is calculated. def get_prune_threshold(model_list, percent): bn_weights = gather_bn_weights(model_list) sorted_bn = torch.sort(bn_weights)[0] # 避免剪掉所有channel的最高阈值(每个BN层的gamma的最大值的最小值即为阈值上限) highest_thre = [] for bnlayer in model_list.values(): highest_thre.append(bnlayer.weight.data.abs().max().item()) highest_thre = min(highest_thre) #...
I tried to train sparsity training on custom dataset. and I want to visualize and check distribution of gamma and bias as you noted on Readme like below  is...