tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Use dataclass config for ACT

Browse Source
This commit is contained in:
Alexander Soare
2024-04-15 09:39:23 +01:00
parent 34f00753eb
commit ef4bd9e25c
3 changed files with 83 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_available.py
View File

@@ -18,14 +18,14 @@ from lerobot.common.datasets.xarm import XarmDataset
from lerobot.common.datasets.aloha import AlohaDataset
from lerobot.common.datasets.pusht import PushtDataset
from lerobot.common.policies.act.policy import ActionChunkingTransformerPolicy
from lerobot.common.policies.act.modeling_act import ActPolicy
from lerobot.common.policies.diffusion.policy import DiffusionPolicy
from lerobot.common.policies.tdmpc.policy import TDMPCPolicy
def test_available():
policy_classes = [
ActionChunkingTransformerPolicy,
ActPolicy,
DiffusionPolicy,
TDMPCPolicy,
]