TD-MPC2数据集使用指南：如何利用300+预训练模型-尧图网络科技

TD-MPC2数据集使用指南：如何利用300+预训练模型

【免费下载链接】tdmpc2Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"项目地址: https://gitcode.com/gh_mirrors/tdm/tdmpc2

TD-MPC2是一个专注于连续控制任务的强化学习框架，提供了300+预训练模型和多任务学习能力，帮助开发者快速上手复杂的机器人控制与决策任务。本文将详细介绍如何获取、配置和使用这些数据集与模型，让你轻松开启强化学习应用之旅。

📊 数据集概览：300+任务覆盖四大平台

TD-MPC2的数据集包含来自四大主流控制平台的任务集合，覆盖从简单机械臂到复杂人形机器人的控制场景：

DMControl：39个基础控制任务（如机械臂抓取、双足行走）
Meta-World：50个多任务操作场景（如按钮按压、抽屉开关）
ManiSkill2：5个高精度操作任务（如立方体堆叠、YCB物体拾取）
MyoSuite：10个生物力学模拟任务（如手部姿态控制、钥匙旋转）

所有数据集均提供标准化评分指标，方便不同算法间的性能对比。通过预训练模型，你可以直接复用在这些任务上训练好的策略，无需从零开始训练。

图：TD-MPC2与其他算法在多任务和单任务场景下的性能对比（越高表示控制精度越好）

🚀 快速开始：三步获取数据集

1️⃣ 克隆项目仓库

首先通过Git获取项目源码：

git clone https://gitcode.com/gh_mirrors/tdm/tdmpc2 cd tdmpc2

2️⃣ 下载预训练数据集

项目提供了两个主要数据集下载脚本，位于datasets/目录：

MT30数据集（轻量级，30个精选任务）：

bash datasets/download_mt30.sh

MT80数据集（全量，80个多平台任务）：

bash datasets/download_mt80.sh

⚠️ 注意：全量数据集大小约15GB，建议根据需求选择合适的数据集。下载完成后数据将自动存放在datasets/目录下。

3️⃣ 配置环境依赖

使用Docker快速配置运行环境：

cd docker docker build -t tdmpc2 . docker run -it --gpus all tdmpc2

或通过conda手动配置（环境配置文件：docker/environment.yaml）：

conda env create -f docker/environment.yaml conda activate tdmpc2

💻 模型使用：一行代码运行预训练模型

TD-MPC2提供了简单直观的模型调用接口，位于tdmpc2/evaluate.py。以下是常用场景的使用示例：

单任务评估

# 评估机械狗跑步任务 python evaluate.py task=dog-run checkpoint=/path/to/dog-1.pt save_video=true

多任务评估

# 评估MT80数据集（48M参数模型） python evaluate.py task=mt80 model_size=48 checkpoint=/path/to/mt80-48M.pt # 评估MT30数据集（317M参数模型） python evaluate.py task=mt30 model_size=317 checkpoint=/path/to/mt30-317M.pt

自定义配置

通过修改tdmpc2/config.yaml文件调整评估参数：

checkpoint：指定预训练模型路径
model_size：选择模型规模（5M/19M/48M/317M）
save_video：是否保存控制过程视频（true/false）

📈 模型选择指南：不同规模模型的应用场景

TD-MPC2提供四种不同参数规模的预训练模型，满足不同需求：

模型规模	参数大小	适用场景	推荐任务类型
5M	轻量级	边缘设备部署	简单控制任务（如摆杆平衡）
19M	标准型	常规研究实验	中等复杂度任务（如猎豹奔跑）
48M	大型	多任务学习	Meta-World平台50任务集合
317M	超大型	高精度控制	复杂操作任务（如双手协作）