iT邦幫忙

save model相關文章
共有 1 則文章
鐵人賽 AI & Data DAY 18

技術 Day 18 存取 Pytroch 訓練好的模型和權重

今天先介紹怎麼把模型存下來,神經網路在訓練的時候所有權重都是存在記憶體中的,只要程式或電腦關掉這些資料就會消失,因此我們必須將這些訓練好的資訊存到硬碟中。 Py...