waxiong

Results 1 comments of waxiong

I also had the same problem,when running python train_with_pruning.py --data coco128.yaml --cfg yolov5s.yaml --weights '' --batch-size 64 RuntimeError: Given groups=1, weight of size [255, 128, 1, 1], expected input[8, 77,...