最近使用tf.keras训练模型，发现我的本地存储权重的模块总是两个.ckpt-data-0000-of-0001和ckpt-data-0001-of-0002，但是tensorflow官方的说法是在多个机器上训练才会有多个，但是我本地一台机器为什么有两个呢？
stackoverflow上看到原因，原来由于我本地模型一部分在cpu，一部分在gpu，最终就产生了两个权重文件，当我禁用gpu后果然只有一个权重文件了。stackoverflow链接Tensorflow checkpoints saving data file
版权声明：本文为qq_40836518原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。