hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Small fix, Refactor diffusion, Diffusion runs (TODO: remove normalization in diffusion)

Browse Source
This commit is contained in:
Remi Cadene
2024-03-02 17:04:39 +00:00
parent 45b4ecb727
commit 80785f8d0e
6 changed files with 449 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/policies/factory.py
View File

@@ -4,7 +4,7 @@ def make_policy(cfg):
policy = TDMPC(cfg.policy)
elif cfg.policy.name == "diffusion":
from lerobot.common.policies.diffusion import DiffusionPolicy
from lerobot.common.policies.diffusion.policy import DiffusionPolicy
policy = DiffusionPolicy(
cfg=cfg.policy,