Browse Source

[MNT] Modify minimum reduced set size

tags/v0.3.2
Gene 2 years ago
parent
commit
670a0fa1b7
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      learnware/specification/utils.py

+ 1
- 1
learnware/specification/utils.py View File

@@ -81,7 +81,7 @@ def generate_rkme_spec(
# Check reduced_set_size
max_reduced_set_size = C.max_reduced_set_size
if reduced_set_size * X[0].size > max_reduced_set_size:
reduced_set_size = max(1, max_reduced_set_size // X[0].size)
reduced_set_size = max(20, max_reduced_set_size // X[0].size)
# Check cuda_idx
if not torch.cuda.is_available() or cuda_idx == -1:


Loading…
Cancel
Save