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

Make sure targets are normalized too (#106)

Browse Source
This commit is contained in:
Alexander Soare
2024-04-26 11:18:39 +01:00
committed by GitHub
parent b980c5dd9e
commit 45f351c618
8 changed files with 116 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/policies/act/configuration_act.py
View File

@@ -75,13 +75,13 @@ class ActionChunkingTransformerConfig:
)
# Normalization / Unnormalization
normalize_input_modes: dict[str, str] = field(
input_normalization_modes: dict[str, str] = field(
default_factory=lambda: {
"observation.image": "mean_std",
"observation.state": "mean_std",
}
)
unnormalize_output_modes: dict[str, str] = field(
output_normalization_modes: dict[str, str] = field(
default_factory=lambda: {
"action": "mean_std",
}