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

Move make_policy_config

Browse Source
This commit is contained in:
Simon Alibert
2025-03-08 12:42:43 +01:00
parent 00698305a3
commit 57e0bdb491
5 changed files with 11 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/scripts/save_policy_to_safetensors.py
View File

@@ -22,10 +22,10 @@ from safetensors.torch import save_file
from lerobot.common.datasets.factory import make_dataset
from lerobot.common.optim.factory import make_optimizer_and_scheduler
from lerobot.common.policies import make_policy
from lerobot.common.policies.factory import make_policy_config
from lerobot.common.utils.random_utils import set_seed
from lerobot.configs.default import DatasetConfig
from lerobot.configs.train import TrainPipelineConfig
from tests.utils import make_policy_config
def get_policy_stats(ds_repo_id: str, policy_name: str, policy_kwargs: dict):