Energy OOD 项目使用教程

1. 项目的目录结构及介绍

energy_ood/
├── CIFAR/
│   ├── snapshots/
│   └── ...
├── demo_figs/
├── utils/
├── LICENSE
├── README.md
└── ...

• CIFAR/: 包含预训练模型和数据集的文件夹。
  • snapshots/: 存放预训练模型的文件夹。
• demo_figs/: 存放演示图表的文件夹。
• utils/: 包含项目中使用的工具函数和脚本。
• LICENSE: 项目的开源许可证文件。
• README.md: 项目的介绍和使用说明文件。

2. 项目的启动文件介绍

项目的启动文件主要是用于运行能量得分测试和训练的脚本。以下是主要的启动文件及其功能：

• run.sh: 用于运行能量得分测试的脚本。
  • bash run.sh energy 0: 运行 CIFAR-10 WRN 的能量得分测试。
  • bash run.sh energy 1: 运行 CIFAR-100 WRN 的能量得分测试。
• run_ft.sh: 用于运行能量得分训练和测试的脚本。
  • bash run_ft.sh energy_ft 0: 运行 CIFAR-10 WRN 的能量得分训练和测试。
  • bash run_ft.sh energy_ft 1: 运行 CIFAR-100 WRN 的能量得分训练和测试。

3. 项目的配置文件介绍

项目中没有明确的配置文件，但可以通过修改启动脚本中的参数来调整模型的行为。例如，可以在 run.sh 或 run_ft.sh 中修改数据集路径、模型路径等参数。

# 示例：修改数据集路径
DATASET_PATH="/path/to/your/dataset"

通过这种方式，可以根据具体需求调整项目的配置。