You are working on the allenai/rslearn repository. The source code is checked out at /testbed at commit…

You are working on the allenai/rslearn repository. The source code is checked out at /testbed at commit…: a task in SWE-Lego-RL-2699 (Harbor dataset). state dict = torch.load("checkpoint.ckpt") state dict "state dict" = { k.replace("model.decoders.segment.", "model.decoders.class."): v for k, v in…

The task

state_dict = torch.load("checkpoint.ckpt") state_dict["state_dict"] = { k.replace("model.decoders.segment.", "model.decoders.class."): v for k, v in state_dict["state_dict"].items() } torch.save(state_dict, "checkpoint.ckpt") ```

Part of Lego-X/Lego-RL-2699.