Browse Source

[FIX] fix bugs about batch_size

tags/v0.3.2
Gene 2 years ago
parent
commit
a9c443aefb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      learnware/market/heterogeneous/organizer/hetero_map/__init__.py

+ 1
- 1
learnware/market/heterogeneous/organizer/hetero_map/__init__.py View File

@@ -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]


Loading…
Cancel
Save