Wendy commited on
Commit
e73817e
·
verified ·
1 Parent(s): cd3a7b0

Upload cogagent_infer_batch.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. cogagent_infer_batch.py +1 -1
cogagent_infer_batch.py CHANGED
@@ -141,7 +141,7 @@ def main():
141
  data_loader = DataLoader(dataset, batch_size=16, shuffle=False)
142
 
143
  res = []
144
- for x in tqdm(dataloader,desc="Processing items"):
145
  #x = data[i]
146
  img_path,task = x
147
  image = []
 
141
  data_loader = DataLoader(dataset, batch_size=16, shuffle=False)
142
 
143
  res = []
144
+ for x in tqdm(data_loader,desc="Processing items"):
145
  #x = data[i]
146
  img_path,task = x
147
  image = []