未验证 提交 cc2edc5e 编写于 作者: J Jiaqi Liu 提交者: GitHub

Update EarlyStopping sample code (#29723) (#29727)

* update EarlyStopping doc

* update EarlyStopping doc, test=document_fix
上级 df0430dc
...@@ -714,7 +714,7 @@ class EarlyStopping(Callback): ...@@ -714,7 +714,7 @@ class EarlyStopping(Callback):
from paddle.vision.models import LeNet from paddle.vision.models import LeNet
from paddle.vision.datasets import MNIST from paddle.vision.datasets import MNIST
from paddle.metric import Accuracy from paddle.metric import Accuracy
from paddle.nn.layer.loss import CrossEntropyLoss from paddle.nn import CrossEntropyLoss
import paddle.vision.transforms as T import paddle.vision.transforms as T
device = paddle.set_device('cpu') device = paddle.set_device('cpu')
...@@ -772,7 +772,8 @@ class EarlyStopping(Callback): ...@@ -772,7 +772,8 @@ class EarlyStopping(Callback):
self.best_weights = None self.best_weights = None
self.stopped_epoch = 0 self.stopped_epoch = 0
self.save_best_model = save_best_model self.save_best_model = save_best_model
self.save_dir = None # `save_dir` is get from `config_callbacks` # The value of `save_dir` is set in function `config_callbacks`
self.save_dir = None
if mode not in ['auto', 'min', 'max']: if mode not in ['auto', 'min', 'max']:
warnings.warn('EarlyStopping mode %s is unknown, ' warnings.warn('EarlyStopping mode %s is unknown, '
'fallback to auto mode.' % mode) 'fallback to auto mode.' % mode)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册