Tianxiaomo
How to continue train according to the last weights(.pth) file?
I trained 100 epochs and get the .pth file, but I want to know how to continue training according the .pth file. Thank you for your help
did you find any way to continue training according to the last weight file
아래 링크 속 line 76 부터 이미 구현된 코드를 참고하여 해결하였습니다.
https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/master/train.py
models.py에서 Yolov4 클래스의 이미 작성된 기능에 오버라이딩 하였으나 동작하지 않아 train.py에서 초기 모델 로딩 후 가중치를 로드하도록 수정하였습니다.
정상적으로 작동합니다!🎉
I tried writing in English, Sorry! 😂
It was solved by referring to the code already implemented from line 76 in the link below.
https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/master/train.py
In models.py, I overridden the already written function of the Yolov4 class, but it did not work, so I modified it to load the weights after loading the initial model in train.py.
Works fine!🎉
