Don't permute r_0 (kitchen) because it's special: required location f…

…or prepare meal
gstrazds · Apr 24, 2023 · 0706968 · 0706968
1 parent 37abd74
commit 0706968
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mingpt/pthru_dataset.py b/mingpt/pthru_dataset.py
@@ -767,7 +767,7 @@ def _tokenize_text(data: dict):
 
     def get_room_tokens(self):
         tokens_list = []
-        for i in range(20):
+        for i in range(1,20):  # leave r_0 (kitchen) alone, because it's special (cooking location, for prepare meal)
             tok_id = self.tokenizer.convert_tokens_to_ids(f"r_{i}")
             if tok_id is not None and tok_id != self.tokenizer.unk_token_id:
                 tokens_list.append(tok_id)