Accelerated FSRS Benchmark

Uses the GPU to accelerate srs-benchmark for FSRS.

Requirements

• Linux or Windows with WSL, and Docker.
• A CUDA device with compute capability 8.6 or newer.

Quickstart

Build the Docker image.

docker compose build srs-benchmark

Prepare the dataset.

docker compose --progress quiet run --rm srs-benchmark python -m src.prepare.prepare --processes 10

If you encounter issues on WSL when preparing the dataset such as concurrent.futures.process.BrokenProcessPool, try one of:

• Increase the memory limit with .wslconfig: https://learn.microsoft.com/windows/wsl/wsl-config
• Use a lower --processes number.

Run training and evaluation.

docker compose --progress quiet run --rm srs-benchmark bash src/main/run.sh