yolact-pytorch icon indicating copy to clipboard operation
yolact-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

百度空間的預訓練權重檔案格式是.h5

Open dt54123 opened this issue 3 years ago • 1 comments

請問大神: 百度空間的預訓練權重檔案格式是.h5，好像不是pytorch通常使用的.pth? 不知道是不是這個原因，導致在載入預訓練模型時跳出『UnpicklingError: invalid load key, 'H'.』的訊息

需要將train.py改成 model_path = ' ' pretrained = False 才能正常運行

dt54123 avatar Jan 25 '22 06:01 dt54123

上传了

bubbliiiing avatar Jan 26 '22 15:01 bubbliiiing