From a9c443aefbbc42c34923922e72515bd92ef83450 Mon Sep 17 00:00:00 2001 From: Gene Date: Thu, 7 Dec 2023 21:04:39 +0800 Subject: [PATCH] [FIX] fix bugs about batch_size --- learnware/market/heterogeneous/organizer/hetero_map/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/learnware/market/heterogeneous/organizer/hetero_map/__init__.py b/learnware/market/heterogeneous/organizer/hetero_map/__init__.py index e124bcb..aca3d02 100644 --- a/learnware/market/heterogeneous/organizer/hetero_map/__init__.py +++ b/learnware/market/heterogeneous/organizer/hetero_map/__init__.py @@ -309,7 +309,7 @@ class HeteroMap(nn.Module): output_feas_list = [] if eval_batch_size * x_test.shape[1] > self.max_process_size: - eval_batch_size = int(self.max_process_size / x_test.shape[1]) + eval_batch_size = max(1, self.max_process_size // x_test.shape[1]) for i in range(0, len(x_test), eval_batch_size): bs_x_test = x_test.iloc[i : i + eval_batch_size]