tensorflow中callbacks

tensorflow中callbacks,第1张

tf.keras.callbacks是回调函数,在训练过程中可以访问用来做一些决策
  1. tf.keras.callbacks.EarlyStopping早停函数,当监控目标monitor训练patience轮都不下降时,结束训练
tf.keras.callbacks.EarlyStopping(
    monitor='val_loss',
    min_delta=0,
    patience=0,
    verbose=0,
    mode='auto',
    baseline=None,
    restore_best_weights=False
)

  1. tf.keras.callbacks.LearningRateScheduler学习率衰减函数
    自己实现scheduler函数设置学习率衰减形式
    3. tf.keras.callbacks.ModelCheckpoint每轮训练结束用来保存模型权重
tf.keras.callbacks.ModelCheckpoint(
    filepath,
    monitor='val_loss',
    verbose=0,
    save_best_only=False,
    save_weights_only=False,
    mode='auto',
    save_freq='epoch',
    options=None,
    initial_value_threshold=None,
    **kwargs
)

  1. tf.keras.callbacks.TensorBoard可视化训练过程
tf.keras.callbacks.TensorBoard(
    log_dir='logs',
    histogram_freq=0,
    write_graph=True,
    write_images=False,
    write_steps_per_second=False,
    update_freq='epoch',
    profile_batch=0,
    embeddings_freq=0,
    embeddings_metadata=None,
    **kwargs
)

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/langs/799345.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-06
下一篇 2022-05-06

发表评论

登录后才能评论

评论列表(0条)

保存