mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 20:19:43 +00:00
new changes
This commit is contained in:
@@ -1,5 +1,5 @@
|
|||||||
lerobot-eval \
|
lerobot-eval \
|
||||||
--policy.path="/raid/jade/models/xvla-libero-og_migrated" \
|
--policy.path="/raid/jade/models/xvla-libero-new_migrated" \
|
||||||
--env.type=libero \
|
--env.type=libero \
|
||||||
--env.task=libero_spatial \
|
--env.task=libero_spatial \
|
||||||
--env.action_type=abs \
|
--env.action_type=abs \
|
||||||
|
|||||||
@@ -331,7 +331,6 @@ def load_state_dict_with_missing_key_handling(
|
|||||||
# Check for missing keys
|
# Check for missing keys
|
||||||
missing_keys = load_result.missing_keys
|
missing_keys = load_result.missing_keys
|
||||||
unexpected_keys = load_result.unexpected_keys
|
unexpected_keys = load_result.unexpected_keys
|
||||||
|
|
||||||
# Filter out whitelisted missing keys
|
# Filter out whitelisted missing keys
|
||||||
policy_type_lower = policy_type.lower()
|
policy_type_lower = policy_type.lower()
|
||||||
whitelisted_keys = known_missing_keys_whitelist.get(policy_type_lower, [])
|
whitelisted_keys = known_missing_keys_whitelist.get(policy_type_lower, [])
|
||||||
|
|||||||
Reference in New Issue
Block a user