Open dt54123 opened 2 years ago
請問大神: 百度空間的預訓練權重檔案格式是.h5,好像不是pytorch通常使用的.pth? 不知道是不是這個原因,導致在載入預訓練模型時跳出『UnpicklingError: invalid load key, 'H'.』的訊息
需要將train.py改成 model_path = ' ' pretrained = False 才能正常運行
上传了
請問大神: 百度空間的預訓練權重檔案格式是.h5,好像不是pytorch通常使用的.pth? 不知道是不是這個原因,導致在載入預訓練模型時跳出『UnpicklingError: invalid load key, 'H'.』的訊息
需要將train.py改成 model_path = ' ' pretrained = False 才能正常運行