在使用自定义模型时,无论设置save_weights_only为True或False,callback时都只保存权重,而不保存模型,是因为检测到模型结构不是Sequential时,会自动将save_weights_only设置为True,导致只保存权重,所以我在源代码调用_save_model时,手动将save_weights_only设置为False,从而实现callback保存模型。
tensorflow2中的tf.keras.callbacks.ModelCheckpoint保存模型问题
最新推荐文章于 2023-03-14 18:35:48 发布