What is the --prune 0 and --prune 1 stands for when doing sparsity training using train.py?

TNTWEN / Pruned-OpenVINO-YOLO

Deploy the pruned YOLOv3/v4/v4-tiny/v4-tiny-3l model on OpenVINO embedded devices

Apache License 2.0

52 stars 11 forks source link

Closed Abhishek672 closed 2 years ago

Abhishek672 commented 2 years ago

Hey @TNTWEN

I am having trouble understanding the above flags. Also does this repository supports the weight pruning during training, similar to https://www.tensorflow.org/model_optimization/guide/pruning/pruning_with_keras ??

Thanks !!