FedML-AI/FedML

OverflowError when use fednlp to train model.

Open

#453 建立於 2022年8月5日

在 GitHub 查看
 (3 留言) (0 反應) (1 負責人)Python (743 fork)batch import
bughelp wantedresearch-projects

倉庫指標

Star
 (3,930 star)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

Hello, I changed model type from "facebook/bart-base" to "facebook/bart-large-cnn". And get this error: image What should I do to slove this problem? Can you give me some tips?

貢獻者指南

研究方向
調查溢出錯誤，比較 bart base 和 bart large cnn 的模型大小；檢查是否需要調整記憶體或批量大小，並查看 fednlp 程式碼庫中是否有硬編碼的張量大小。
技術棧
python
領域
machine learning
議題類型
錯誤
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 小時
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
大致清晰
前置要求
PythonHugging Face Transformers
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
70