Only 6 Images are getting Trained - Sparsity Training

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏

Apache License 2.0

1.5k stars 446 forks source link

Only 6 Images are getting Trained - Sparsity Training #141

Open anusha657 opened 3 years ago

anusha657 commented 3 years ago

Hi ,

Even though there are 20 images but still only 6 images are being detected while performing sparsity Training . Training_Issue

Command Used - python train.py --cfg cfg/yolov4.cfg --data data/objcoco.data --weights /home/anusha/yolov4.weights --epochs 15 --batch-size 8 -sr --s 0.001 --prune 1