tensorflow2中的tf.keras.callbacks.ModelCheckpoint保存模型问题

在使用自定义模型时,无论设置save_weights_only为True或False,callback时都只保存权重,而不保存模型,是因为检测到模型结构不是Sequential时,会自动将save_weights_only设置为True,导致只保存权重,所以我在源代码调用_save_model时,手动将save_weights_only设置为False,从而实现callback保存模型。

你可能感兴趣的:(tensorflow2中的tf.keras.callbacks.ModelCheckpoint保存模型问题)