我想为我的实验保存多个模型,但注意到tf.train.Saver()
构造函数最多只能保存5个模型。以下是一个简单的代码示例:
当我运行这段代码时,我在桌面上只看到了5个模型。这是为什么呢?我怎样才能使用同一个tf.train.Saver()
构造函数保存超过5个模型?
回答:
tf.train.Saver()
构造函数有一个可选参数max_to_keep
,默认值为保存最近的5个检查点。要保存更多的模型,只需为该参数指定一个值即可:
要保留所有检查点,请在构造函数中将max_to_keep
参数设置为None
。