BatchBALD 项目使用教程

BatchBALD 项目使用教程

1. 项目目录结构及介绍

BatchBALD 项目的目录结构如下：

BatchBALD/
├── laaos_results/
│   └── paper/
├── notebooks/
│   └── paper/
├── src/
│   ├── run_experiment.py
│   └── ...
├── LICENSE
├── README.md
├── pyproject.toml
└── requirements.txt

目录结构介绍

• laaos_results/: 包含与论文相关的实验结果文件。
• notebooks/: 包含与论文相关的 Jupyter Notebook 文件。
• src/: 项目的源代码目录，包含主要的 Python 脚本文件。
• LICENSE: 项目的许可证文件，采用 GPL-3.0 许可证。
• README.md: 项目的说明文件，包含项目的基本信息和使用说明。
• pyproject.toml: 项目的配置文件，用于定义项目的构建系统和依赖。
• requirements.txt: 项目的依赖文件，列出了项目运行所需的 Python 包。

2. 项目的启动文件介绍

项目的启动文件是 src/run_experiment.py。该文件用于启动实验，并配置实验的参数。

启动文件介绍

• run_experiment.py: 该脚本用于启动实验。可以通过命令行参数配置实验的各项参数，例如数据集、模型、采样数量等。

示例命令

python src/run_experiment.py --quickquick --num_inference_samples 10 --available_sample_k 40

• --quickquick: 快速启动实验。
• --num_inference_samples: 设置推理样本数量。
• --available_sample_k: 设置可用样本数量。

3. 项目的配置文件介绍

项目的配置文件主要包括 pyproject.toml 和 requirements.txt。

配置文件介绍

• pyproject.toml: 该文件定义了项目的构建系统和依赖。它通常包含项目的基本信息、依赖项、构建工具等。

• requirements.txt: 该文件列出了项目运行所需的 Python 包及其版本。通过运行以下命令可以安装这些依赖项：

pip install -r requirements.txt

依赖安装

在运行项目之前，确保安装了所有依赖项。可以通过以下命令安装依赖：

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
pip install -r requirements.txt

以上命令将安装 PyTorch 和其他必要的依赖项，确保项目能够正常运行。