put meanstd as default

This commit is contained in:
Jade Choghari
2026-01-02 10:27:55 +01:00
parent 508d65a827
commit f0d0faa8a1

View File

@@ -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
}
)