Shortcuts

Installation

  1. Set up the OpenCompass environment:

conda create --name opencompass python=3.10 pytorch torchvision pytorch-cuda -c nvidia -c pytorch -y
conda activate opencompass

If you want to customize the PyTorch version or related CUDA version, please refer to the official documentation to set up the PyTorch environment. Note that OpenCompass requires pytorch>=1.13.

  1. Install OpenCompass:

    git clone https://github.com/open-compass/opencompass.git
    cd opencompass
    pip install -e .
    
  2. Install humaneval (Optional)

    If you want to evaluate your models coding ability on the humaneval dataset, follow this step.

    click to show the details
    git clone https://github.com/openai/human-eval.git
    cd human-eval
    pip install -r requirements.txt
    pip install -e .
    cd ..
    

    Please read the comments in human_eval/execution.py lines 48-57 to understand the potential risks of executing the model generation code. If you accept these risks, uncomment line 58 to enable code execution evaluation.

  3. Install Llama (Optional)

    If you want to evaluate Llama / Llama-2 / Llama-2-chat with its official implementation, follow this step.

    click to show the details
    git clone https://github.com/facebookresearch/llama.git
    cd llama
    pip install -r requirements.txt
    pip install -e .
    cd ..
    

    You can find example configs in configs/models. (example)

  4. Install alpaca-eval (Optional):

    If you want toevaluate alpaca-eval in official ways, follow this step.

    click to show the details
    pip install alpaca-eval
    

Dataset Preparation

The datasets supported by OpenCompass mainly include two parts:

  1. Huggingface datasets: The Huggingface Datasets provide a large number of datasets, which will automatically download when running with this option.

  2. Custom dataset: OpenCompass also provides some Chinese custom self-built datasets. Please run the following command to manually download and extract them.

Run the following commands to download and place the datasets in the ${OpenCompass}/data directory can complete dataset preparation.

# Run in the OpenCompass directory
wget https://github.com/open-compass/opencompass/releases/download/0.2.2.rc1/OpenCompassData-core-20240207.zip
unzip OpenCompassData-core-20240207.zip

If you need to use the more comprehensive dataset (~500M) provided by OpenCompass, You can download and unzip it using the following command:

wget https://github.com/open-compass/opencompass/releases/download/0.2.2.rc1/OpenCompassData-complete-20240207.zip
unzip OpenCompassData-complete-20240207.zip
cd ./data
find . -name "*.zip" -exec unzip "{}" \;

The list of datasets included in both .zip can be found here

OpenCompass has supported most of the datasets commonly used for performance comparison, please refer to configs/dataset for the specific list of supported datasets.

For next step, please read Quick Start.

Read the Docs v: latest
Versions
latest
stable
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.
@沪ICP备2021009351号-23 OpenCompass Open Platform Service Agreement