From ef24d89db1ae9b607955c08b187b146ac0ccf75e Mon Sep 17 00:00:00 2001 From: thodoris Date: Fri, 17 Aug 2018 21:33:41 +0300 Subject: [PATCH] Update base.py changed hard-coded randit upper bound value (4294967295) with np.iinfo(np.int32).max-1. In windows 64bit with Conda , the randint fails with the hard-coded value (4294967295). --- braindecode/models/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/braindecode/models/base.py b/braindecode/models/base.py index 2f865d0..9dbd9eb 100644 --- a/braindecode/models/base.py +++ b/braindecode/models/base.py @@ -148,11 +148,11 @@ class BaseModel(object): self.iterator = CropsFromTrialsIterator( batch_size=batch_size, input_time_length=input_time_length, n_preds_per_input=n_preds_per_input, - seed=self.seed_rng.randint(0, 4294967295)) + seed=self.seed_rng.randint(0, np.iinfo(np.int32).max-1)) else: self.iterator = BalancedBatchSizeIterator( batch_size=batch_size, - seed=self.seed_rng.randint(0, 4294967295)) + seed=self.seed_rng.randint(0, np.iinfo(np.int32).max-1)) stop_criterion = MaxEpochs(epochs - 1)# -1 since we dont print 0 epoch, which matters for this stop criterion train_set = SignalAndTarget(train_X, train_y) optimizer = self.optimizer