close
身為一個門外漢
搞了半個月才跑好範例程式還是蠻丟臉的
總而言之是在練習這個專案 https://www.tensorflow.org/tutorials/text/image_captioning
本身就有保存checkpoint了
但是在重用之前要記得重新叫一下encoder 還有decoder
而且要把train的時候定義的訓練集大小調整成一樣的
不然會造成tensor大小不對的情況
今天弄了半天訓練好的20000筆訓練資料的checkpoint在我自己另外摳定的檔案出現了tensor大小不一樣的情況
後來調整了一下train_captions就可以跑了
我是想說輸入應該跟訓練的資料沒關係才對吧....
文章標籤
全站熱搜
留言列表
碼農 (8)
{{ article.title }}