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

Move tdmpc files

Browse Source
This commit is contained in:
Simon Alibert
2024-03-09 18:44:36 +01:00
parent 1cc621ec36
commit 59397fb44a
4 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,6 +1,6 @@
def make_policy(cfg): def make_policy(cfg):
if cfg.policy.name == "tdmpc": if cfg.policy.name == "tdmpc":
from lerobot.common.policies.tdmpc import TDMPC from lerobot.common.policies.tdmpc.policy import TDMPC
policy = TDMPC(cfg.policy, cfg.device) policy = TDMPC(cfg.policy, cfg.device)
elif cfg.policy.name == "diffusion": elif cfg.policy.name == "diffusion":

0
lerobot/common/policies/tdmpc/__init__.py Normal file
View File

0
lerobot/common/policies/tdmpc_helper.py → lerobot/common/policies/tdmpc/helper.py
View File

2
lerobot/common/policies/tdmpc.py → lerobot/common/policies/tdmpc/policy.py
View File

@@ -8,7 +8,7 @@ import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
import lerobot.common.policies.tdmpc_helper as h import lerobot.common.policies.tdmpc.helper as h
FIRST_FRAME = 0 FIRST_FRAME = 0