mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-31 19:01:28 +00:00
put meanstd as default
This commit is contained in:
@@ -67,8 +67,8 @@ class PI0FastConfig(PreTrainedConfig):
|
||||
normalization_mapping: dict[str, NormalizationMode] = field(
|
||||
default_factory=lambda: {
|
||||
"VISUAL": NormalizationMode.IDENTITY,
|
||||
"STATE": NormalizationMode.QUANTILES, # Pi0Fast uses quantiles for state
|
||||
"ACTION": NormalizationMode.QUANTILES, # Pi0Fast uses quantiles for action
|
||||
"STATE": NormalizationMode.MEAN_STD, # Pi0Fast uses quantiles for state
|
||||
"ACTION": NormalizationMode.MEAN_STD, # Pi0Fast uses quantiles for action
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user