小编Gri*_*yan的帖子

我想使用 GPU 设备训练我的自定义模型。我想知道客户端可以通过 CPU 使用它吗？

我正在训练模型，fit_generator()并希望为每个纪元保存的权重生成唯一的名称

已经尝试过：请参阅后面的代码

代码：

model_path = '.\checkpoints\cp{}.ckpt'.format(time())
cp_callback = tf.keras.callbacks.ModelCheckpoint(model_path, 
                                                 verbose=1,
                                                  period=2)

model.fit_generator(..........,callbacks=[cp_callback])

预期：生成唯一的检查点名称，
例如 epoch_4.ckpt 或 epoch_5.ckpt
实际：每次保存时，都会覆盖现有检查点