WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.86k stars 585 forks source link

yolov4-tiny 預訓練權重 #407

Open chris-ym opened 2 years ago

chris-ym commented 2 years ago

您好, 我想使用darknet中的yolov4-tiny之權重(yolov4-tiny.conv.29做為我的預訓練權重,再訓練我自己的資料,我使用load_darknet_weight有成功載入權重檔案,但訓練一開始就抓不到cls 、obj 、box、total等都為nan,請問這個問題該如何解決?