fix: batch size arg

violenil · violenil · commit 677268875270 · 2024-09-18T10:10:13.000+02:00
diff --git a/run_chunked_eval.py b/run_chunked_eval.py
@@ -74,7 +74,7 @@ def main(model_name, strategy, task_name):
         output_folder='results-chunked-pooling',
         eval_splits=['test'],
         overwrite_results=True,
-        encode_kwargs = {'batch_size': batch_size},
+        encode_kwargs = {'batch_size': 1},
     )
 
     tasks = [

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ def main(model_name, strategy, task_name):`
`74`	`74`	`output_folder='results-chunked-pooling',`
`75`	`75`	`eval_splits=['test'],`
`76`	`76`	`overwrite_results=True,`
`77`		`- encode_kwargs = {'batch_size': batch_size},`
	`77`	`+ encode_kwargs = {'batch_size': 1},`
`78`	`78`	`)`
`79`	`79`
`80`	`80`	`tasks = [`