hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

backup wip

Browse Source
This commit is contained in:
Alexander Soare
2024-04-05 11:03:28 +01:00
parent 3a4dfa82fe
commit edb125b351
3 changed files with 188 additions and 213 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/convert_act_weights.py
View File

@@ -42,6 +42,8 @@ start_replacements = [
("model.transformer.encoder.", "model.encoder."),
("model.transformer.decoder.", "model.decoder."),
("model.backbones.0.0.body.", "model.backbone."),
("model.additional_pos_embed.weight", "model.encoder_robot_and_latent_pos_embed.weight"),
("model.cls_embed.weight", "model.vae_encoder_cls_embed.weight"),
]
for to_replace, replace_with in start_replacements: