From 36d11b364238e20d08d15b624c0b37ca5785cd5c Mon Sep 17 00:00:00 2001 From: Oleg Klimov Date: Thu, 9 May 2024 08:54:55 +0200 Subject: [PATCH] repeat batch 2 times --- self_hosting_machinery/inference/inference_worker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/self_hosting_machinery/inference/inference_worker.py b/self_hosting_machinery/inference/inference_worker.py index 0802eeb2..64d41d1f 100644 --- a/self_hosting_machinery/inference/inference_worker.py +++ b/self_hosting_machinery/inference/inference_worker.py @@ -73,7 +73,7 @@ def check_cancelled(*args, **kwargs): return set() log("STATUS test batch") - for _ in range(10): + for _ in range(2): inference_model.infer(dummy_call, DummyUploadProxy, {}) if compile: return