From 2ec3c0711a9719ac324388adc662e471f8fe2168 Mon Sep 17 00:00:00 2001 From: generatedunixname537391475639613 Date: Tue, 3 Feb 2026 13:14:44 -0800 Subject: [PATCH] fbcode/deeplearning/projects/sam3_release/sam3/train/data Reviewed By: JuanBesa Differential Revision: D91210167 fbshipit-source-id: a563232f4bc82f6f3b99e53df1c88cf0f39747bb --- sam3/train/data/collator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sam3/train/data/collator.py b/sam3/train/data/collator.py index 4a0f2e8..38d031d 100644 --- a/sam3/train/data/collator.py +++ b/sam3/train/data/collator.py @@ -194,7 +194,7 @@ def collate_fn_api( offset_img_id = 0 offset_query_id = [0 for _ in range(num_stages)] - for i, data in enumerate(batch): + for data in batch: img_batch.extend([img.data for img in data.images]) if data.raw_images is not None: @@ -209,7 +209,7 @@ def collate_fn_api( datapoint_query_id_2_stage_query_id.append(offset_query_id[stage_id]) offset_query_id[stage_id] += 1 - for j, q in enumerate(data.find_queries): + for q in data.find_queries: stage_id = q.query_processing_order stages[stage_id].img_ids.append(q.image_id + offset_img_id) if q.query_text not in text_batch: