search

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.87k stars 585 forks source link

UnboundLocalError: local variable 'epoch' referenced before assignment #15

Open mary-0830 opened 4 years ago

mary-0830 commented 4 years ago

Traceback (most recent call last): File "/home/jjliao/code/PyTorch_yolov4/train.py", line 415, in train() # train normally File "/home/jjliao/code/PyTorch_yolov4/train.py", line 373, in train print('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600)) UnboundLocalError: local variable 'epoch' referenced before assignment

请问，这个epoch是没有定义嘛？这个需要加全局变量嘛？求解答，谢谢！

WongKinYiu commented 4 years ago

you can add epoch=0 before this line. https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/master/train.py#L222

mary-0830 commented 4 years ago

thanks.

ldfinfontainebleau commented 3 years ago

epoch > 299 will be fine if you use pretrained weight.