最近使用tf.keras训练模型,发现我的本地存储权重的模块总是两个.ckpt-data-0000-of-0001和ckpt-data-0001-of-0002,但是tensorflow官方的说法是在多个机器上训练才会有多个,但是我本地一台机器为什么有两个呢?
stackoverflow上看到原因,原来由于我本地模型一部分在cpu,一部分在gpu,最终就产生了两个权重文件,当我禁用gpu后果然只有一个权重文件了。stackoverflow链接Tensorflow checkpoints saving data file
ckpt-data-xxxx-of-xxxx有两个
最新推荐文章于 2023-08-23 16:58:19 发布