Restore max_iter behavior during training

Teichlab · Aug 11, 2023 · b3ce291 · b3ce291
1 parent 83cf73e
commit b3ce291
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/celltypist/train.py b/celltypist/train.py
@@ -355,9 +355,7 @@ def train(X = None,
  indata = indata.toarray()
  #max_iter
  if max_iter is None:
- if use_GPU:
- max_iter = 1000
- elif indata.shape[0] < 50000:
+ if indata.shape[0] < 50000:
  max_iter = 1000
  elif indata.shape[0] < 500000:
  max_iter = 500