Merge pull request #9 from ericjang/patch-1

remove duplicated model reload
octo-models · Dec 14, 2023 · 37951e4 · 37951e4
2 parents f5d5b08 + f86f1cb
commit 37951e4
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/examples/02_finetune_new_observation_action.py b/examples/02_finetune_new_observation_action.py
@@ -105,7 +105,6 @@ def process_batch(batch):
 
     # load pre-training config and modify --> remove wrist cam, add proprio input, change action head
     # following Zhao et al. we use "action chunks" of length 50 and L1 loss for ALOHA
-    pretrained_model = OctoModel.load_pretrained(FLAGS.pretrained_path)
     config = pretrained_model.config
     del config["model"]["observation_tokenizers"]["wrist"]
     ###