Add training on custom openpi datasets

Cleanup instructions

clean up doc

pass linter

updates

Add test
This commit is contained in:
Michael Equi
2024-12-22 19:19:54 +00:00
parent 385780ecc3
commit 9da84a2f7f
9 changed files with 75 additions and 46 deletions

View File

@@ -16,6 +16,7 @@ from openpi.policies import libero_policy
from openpi.policies import policy as _policy
from openpi.policies import policy_config as _policy_config
from openpi.serving import websocket_policy_server
from openpi.shared import delta_actions
from openpi.training import config as _config
@@ -146,7 +147,7 @@ def create_default_policy(
logging.info("Creating policy...")
match env:
case EnvMode.ALOHA:
delta_action_mask = _policy_config.make_bool_mask(6, -1, 6, -1)
delta_action_mask = delta_actions.make_bool_mask(6, -1, 6, -1)
config = make_policy_config(
input_layers=[
aloha_policy.ActInputsRepack(),