Callback--EarlyStopping

EarlyStopping

参数详解

参数名称 含义 默认值
monitor 要监视的指标(log()记录的指标) None
min_delta 最小误差 0.0
patience monitor不在变化时再检查k次后停止训练 3
verbose 详细模式 False
mode 监视指标的最大值还是最小值.对于loss应使用min,对于accuracy应使用max ‘min’
strict 当log()中不存在monitor时停止训练 True
check_finite monitor变成NaN或无穷时停止训练 True
stopping_threshold monitor达到阈值立即停止训练 None
divergence_threshold 一旦monitor变得比阈值更差就停止训练 Noe
check_on_train_epoch_end 在每个training_epoch_end后执行检查,否则在val_epoch_end后检查 None(True)

你可能感兴趣的:(深度学习,pytorch,机器学习)