# Early Stopping for PyTorch
Early stopping is a form of regularization used to avoid overfitting on the training dataset. Early stopping keeps track of the validation loss, if the loss stops decreasing for several epochs in a row the training stops. The ```EarlyStopping``` class in ```pytorchtool.py``` is used to create an object to keep track of the validation loss while training a [PyTorch](https://pytorch.org/) model. It will save a checkpoint of the model each time the validation loss decrease. We set the ```patience``` argument in the ```EarlyStopping``` class to how many epochs we want to wait after the last time the validation loss improved before breaking the training loop. There is a simple example of how to use the ```EarlyStopping``` class in the [MNIST_Early_Stopping_example](MNIST_Early_Stopping_example.ipynb) notebook.
Underneath is a plot from the example notebook, which shows the last checkpoint made by the EarlyStopping object, right before the model started to overfit. It had patience set to 20.
![Loss plot](loss_plot.png?raw=true)
## Usage
You can run this project directly in the browser by clicking this button: [![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/Bjarten/early-stopping-pytorch/master), or you can clone the project to your computer and install the required pip packages specified in the requirements text file.
```
pip install -r requirements.txt
```
## References
The ```EarlyStopping``` class in ```pytorchtool.py``` is inspired by the [ignite EarlyStopping class](https://github.com/pytorch/ignite/blob/master/ignite/handlers/early_stopping.py).
没有合适的资源?快使用搜索试试~ 我知道了~
early-stopping-pytorch:提前停止PyTorch
共9个文件
md:2个
png:1个
license:1个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
5星 · 超过95%的资源 21 下载量 200 浏览量
2021-05-10
15:34:24
上传
评论 1
收藏 517KB ZIP 举报
温馨提示
提前停止PyTorch 提早停止是一种正则化形式,用于避免过度拟合训练数据集。 提前停止跟踪验证损失,如果损失连续连续几个时期停止减少,则训练将停止。 该EarlyStopping类pytorchtool.py用于创建一个对象来跟踪验证损失的同时培养了模型。 每当验证损失减少时,它将保存模型的检查点。 我们在EarlyStopping类中将patience参数设置为在上次验证损失改善后要中断训练循环之前我们要等待多少个时期。 在笔记本中,有一个简单的示例说明如何使用EarlyStopping类。 下面是示例笔记本的图,该图显示了在模型开始过拟合之前,EarlyStopping对象所做的最后一个检查点。 耐心设定为20。 用法 您可以通过单击以下按钮直接在浏览器中运行该项目: ,也可以将项目克隆到计算机上,然后安装需求文本文件中指定的必需pip软件包。 pip install -r r
资源推荐
资源详情
资源评论
收起资源包目录
early-stopping-pytorch-master.zip (9个子文件)
early-stopping-pytorch-master
pytorchtools.py 2KB
loss_plot.png 29KB
LICENSE 1KB
MNIST_Early_Stopping_example.ipynb 86KB
requirements.txt 29B
.gitignore 1KB
CODE_OF_CONDUCT.md 3KB
checkpoint.pt 463KB
README.md 2KB
共 9 条
- 1
资源评论
- sweeney5202023-01-08资源内容详细,总结地很全面,与描述的内容一致,对我启发很大,学习了。
- 破茧重生2022-11-11这个资源总结的也太全面了吧,内容详实,对我帮助很大。
- m0_464931732023-07-29资源很受用,资源主总结的很全面,内容与描述一致,解决了我当下的问题。
- chengjianyangshi2022-11-30支持这个资源,内容详细,主要是能解决当下的问题,感谢大佬分享~
管墨迪
- 粉丝: 26
- 资源: 4665
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- Spring Cloud商城项目专栏 049 支付
- sensors-18-03721.pdf
- Facebook.apk
- 推荐一款JTools的call-this-method插件
- json的合法基色来自红包东i请各位
- 项目采用YOLO V4算法模型进行目标检测,使用Deep SORT目标跟踪算法 .zip
- 针对实时视频流和静态图像实现的对象检测和跟踪算法 .zip
- 部署 yolox 算法使用 deepstream.zip
- 基于webmagic、springboot和mybatis的MagicToe Java爬虫设计源码
- 通过实时流协议 (RTSP) 使用 Yolo、OpenCV 和 Python 进行深度学习的对象检测.zip
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功