Upload infer_1.py with huggingface_hub
Browse files- infer_1.py +1 -1
infer_1.py
CHANGED
|
@@ -43,7 +43,7 @@ save_data = []
|
|
| 43 |
correct_num = 0
|
| 44 |
begin = 0
|
| 45 |
end = len(data)
|
| 46 |
-
batch_size =
|
| 47 |
for batch_idx in tqdm(range(begin, end, batch_size)):
|
| 48 |
batch = data[batch_idx:batch_idx + batch_size]
|
| 49 |
|
|
|
|
| 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]
|
| 49 |
|