mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-15 08:39:49 +00:00
Add default config for ACT
This commit is contained in:
@@ -117,6 +117,16 @@ def get_default_peft_configuration(policy_type):
|
||||
"unnormalize_outputs",
|
||||
],
|
||||
}
|
||||
elif policy_type == "act":
|
||||
return {
|
||||
"target_modules": r"(.*_proj|.*\.action_head)",
|
||||
"modules_to_save": [
|
||||
# These are inf on load otherwise
|
||||
"normalize_inputs",
|
||||
"normalize_targets",
|
||||
"unnormalize_outputs",
|
||||
],
|
||||
}
|
||||
|
||||
return {'modules_to_save': None}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user