Wendy-Fly commited on
Commit
aa7c676
·
verified ·
1 Parent(s): 0fd5fe3

Upload infer__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. infer__.py +1 -1
infer__.py CHANGED
@@ -42,7 +42,7 @@ data = read_json('/home/zbz5349/WorkSpace/aigeeks/Qwen2.5-VL/LLaMA-Factory/data/
42
  save_data = []
43
  correct_num = 0
44
  begin = 0
45
- end = 1
46
  batch_size = 1
47
  for batch_idx in tqdm(range(begin, end, batch_size)):
48
  batch = data[batch_idx:batch_idx + batch_size]
 
42
  save_data = []
43
  correct_num = 0
44
  begin = 0
45
+ end = len(data)
46
  batch_size = 1
47
  for batch_idx in tqdm(range(begin, end, batch_size)):
48
  batch = data[batch_idx:batch_idx + batch_size]