ckpt-data-xxxx-of-xxxx有兩個

最近使用tf.keras訓練模型,發現我的本地存儲權重的模塊總是兩個.ckpt-data-0000-of-0001和ckpt-data-0001-of-0002,但是tensorflow官方的說法是在多個機器上訓練纔會有多個,但是我本地一臺機器爲什麼有兩個呢? stackoverflow上看到原因,原來由於我本地模型一部分在cpu,一部分在gpu,最終就產生了兩個權重文件,當我禁用gpu後果然只有
相關文章
相關標籤/搜索