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

bubbliiiing / yolact-pytorch

这是一个yolact-pytorch的库，可用于训练自己的数据集

MIT License

70 stars 12 forks source link

Open dt54123 opened 2 years ago

dt54123 commented 2 years ago

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

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

bubbliiiing commented 2 years ago

上传了