RuntimeError: Error(s) in loading state_dict for TransformerMASSModel:解決方法

運行MASS/MASS-summarization代碼報錯:RuntimeError: Error(s) in loading state_dict for TransformerMASSModel: 因爲fine-tuning階段需要加載之前訓練好得Pre-training模型,在之前訓練好得預訓練模型中包含model,fine-tuning階段不需要,所以去掉就可以,或者在預訓練階段去掉mod
相關文章
相關標籤/搜索