diff --git a/solo/commands/robots/lerobot/recording.py b/solo/commands/robots/lerobot/recording.py index a302375..efe4278 100644 --- a/solo/commands/robots/lerobot/recording.py +++ b/solo/commands/robots/lerobot/recording.py @@ -1131,6 +1131,10 @@ def training_mode(config: dict): wandb=wandb_config, seed=1000, resume=resume_training, # Use the resume flag we determined above + rename_map={ + "observation.images.side": "observation.images.camera1", + "observation.images.up": "observation.images.camera2" + }, ) typer.echo("🎓 Starting training... This may take a while.")