AI开发平台MODELARTS-构建学习器:基于learner进行模型评估

时间：2024-05-28 16:22:36

learner.validate(
	data=db,
    checkpoint="your_local_path_of_pretrained_model",
    gpu_ids=[0]
)

表3 learner.validate参数
参数名称	可选/必选	参数类型	参数描述
data	可选	DataBlock object	数据集对象，默认为None，初始化方法参考DataBlock模块。默认值时会基于构建好的DataBlock对象中的validation data进行模型评估。
checkpoint	可选	string	预训练模型路径，默认为None。当基于learner.fit完成训练且该参数为None，则基于训练后的模型参数进行评估。如果指定checkpoint路径，则加载对应路径的模型参数进行评估。
gpu_ids	可选	int/list	模型评估时使用的GPU序号。open-mmlab系列模型默认使用0号卡。ivg系列可指定，如0或者[0, 1]。
show_score_thr	可选	float	评估时预测结果置信度阈值，默认为0.5，仅用于open-mmlab系列模型。
out	可选	string	评估结果pkl文件保存路径，默认为None，仅用于open-mmlab系列模型。