opencompass
opencompass copied to clipboard
Switch to python runner for single GPU
Motivation
There is no need to use torchrun
for single GPU inference.
Besides, python
runner is more friendly for debugging.
The debugging snippet is as follows
https://github.com/open-compass/opencompass/blob/889e7e11409d83fe312ecc7d7f0ed8861a84cc92/opencompass/runners/local.py#L116-L131