diff --git a/eval.sh b/eval.sh index 37b67ad3b..c4ce178e4 100644 --- a/eval.sh +++ b/eval.sh @@ -1,5 +1,5 @@ lerobot-eval \ - --policy.path="/raid/jade/models/xvla-libero-og_migrated" \ + --policy.path="/raid/jade/models/xvla-libero-new_migrated" \ --env.type=libero \ --env.task=libero_spatial \ --env.action_type=abs \ diff --git a/src/lerobot/processor/migrate_policy_normalization.py b/src/lerobot/processor/migrate_policy_normalization.py index 525b7431c..dd7ecba4a 100644 --- a/src/lerobot/processor/migrate_policy_normalization.py +++ b/src/lerobot/processor/migrate_policy_normalization.py @@ -331,7 +331,6 @@ def load_state_dict_with_missing_key_handling( # Check for missing keys missing_keys = load_result.missing_keys unexpected_keys = load_result.unexpected_keys - # Filter out whitelisted missing keys policy_type_lower = policy_type.lower() whitelisted_keys = known_missing_keys_whitelist.get(policy_type_lower, [])