开源项目 `early-stopping-pytorch` 使用教程

开源项目 early-stopping-pytorch 使用教程

early-stopping-pytorchEarly stopping for PyTorch 项目地址:https://gitcode.com/gh_mirrors/ea/early-stopping-pytorch

项目目录结构及介绍

early-stopping-pytorch/
├── README.md
├── early_stopping.py
├── examples/
│   ├── example_1.py
│   └── example_2.py
├── tests/
│   ├── test_early_stopping.py
│   └── __init__.py
└── setup.py

• README.md: 项目说明文档，包含项目的基本介绍、安装方法和使用示例。
• early_stopping.py: 核心文件，包含早期停止机制的实现。
• examples/: 示例文件夹，包含多个使用早期停止机制的示例脚本。
• tests/: 测试文件夹，包含项目的单元测试。
• setup.py: 项目的安装脚本，用于通过 pip 安装项目。

项目的启动文件介绍

项目的启动文件是 early_stopping.py，该文件定义了早期停止机制的核心逻辑。以下是该文件的主要内容：

import numpy as np
import torch

class EarlyStopping:
    def __init__(self, patience=7, verbose=False, delta=0):
        self.patience = patience
        self.verbose = verbose
        self.counter = 0
        self.best_score = None
        self.early_stop = False
        self.val_loss_min = np.Inf
        self.delta = delta

    def __call__(self, val_loss, model):
        score = -val_loss

        if self.best_score is None:
            self.best_score = score
            self.save_checkpoint(val_loss, model)
        elif score < self.best_score + self.delta:
            self.counter += 1
            if self.counter >= self.patience:
                self.early_stop = True
        else:
            self.best_score = score
            self.save_checkpoint(val_loss, model)
            self.counter = 0

    def save_checkpoint(self, val_loss, model):
        if self.verbose:
            print(f'Validation loss decreased ({self.val_loss_min:.6f} --> {val_loss:.6f}).  Saving model ...')
        torch.save(model.state_dict(), 'checkpoint.pt')
        self.val_loss_min = val_loss

该文件定义了一个 EarlyStopping 类，用于在训练过程中实现早期停止机制。类的实例可以被调用，并在满足条件时停止训练。

项目的配置文件介绍

项目中没有显式的配置文件，但可以通过修改 early_stopping.py 中的参数来配置早期停止的行为。例如，可以修改 patience、verbose 和 delta 参数来调整早期停止的策略。

early_stopping = EarlyStopping(patience=10, verbose=True, delta=0.001)

通过这种方式，可以根据具体需求调整早期停止的参数。

early-stopping-pytorchEarly stopping for PyTorch 项目地址:https://gitcode.com/gh_mirrors/ea/early-stopping-pytorch