quantizaion problem

[saeedkhanehgir]

trafficstars

Hi thanks for this project I use command below for training with quantization but get problem.

python train.py -pt --data cfg/obj.data --batch-size 4 --weights weights/yolov3-tiny.weights --cfg cfg/yolov3-tiny.cfg -sr --s .001 --prune 0 --quantized 2

error: File "train.py", line 920, in WarmupForQ(hyp, step=i, a_bit=a_bit, w_bit=w_bit) File "train.py", line 799, in WarmupForQ bn_weights = gather_bn_weights(model.module_list, [idx]) NameError: name 'idx' is not defined