如何在DVC多阶段实验中提取当前运行的阶段名称？

方案一：显式传递阶段标识（最直接可靠） #

修改dvc.yaml的阶段命令，将阶段名和cycle参数拼接后传递给Python脚本：

Stages:
 Training:
   foreach:
    -cycle: 0
    -cycle: 1
    -cycle: 2
   do:
    cmd: python train.py --stage "${stage}_${item.cycle}"
 Selection:
   foreach:
    -cycle: 0
    -cycle: 1
    -cycle: 2
   do:
    cmd: python train.py --stage "${stage}_${item.cycle}"

在train.py中通过命令行参数接收：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--stage', type=str, required=True)
args = parser.parse_args()

# 当前运行的阶段标识，如'Training_0'或'Selection_0'
current_stage = args.stage

方案二：读取DVC环境变量自动解析 #

DVC运行阶段时会自动注入环境变量，可在Python中读取并拼接出目标标识：

import os
import json

# 获取原始阶段名（如'Training'、'Selection'）
base_stage = os.environ.get('DVC_STAGE')
# 读取当前循环的参数（DVC_PARAMS是JSON格式字符串）
params = json.loads(os.environ.get('DVC_PARAMS', '{}'))
cycle = params.get('cycle')

# 拼接成目标格式
current_stage = f"{base_stage}_{cycle}" if cycle else base_stage

补充说明 #

dvc.api主要用于项目级别的数据/参数读取，无法直接获取当前运行阶段的上下文信息，因此需要通过上述传递参数或读取环境变量的方式实现需求。

内容的提问来源于stack exchange，提问作者Razor