StopIteration

[reborm]

StopIteration

[mpyrozhok]

What is your batch size and how many samples do you have per epoch?

[reborm]

OK, I figured it out. The bug happened in other place in my project. Thank you

[paulcx]

@mpyrozhok @reborm I have same error with "StopIteration" at same place. batch_size is 140 and 20580 samples each epoch
Any idea?

[mpyrozhok]

@paulcx This can happen when you use some batch size and epoch size values while initializing scheduler and then supplying different values to your dataloader, or any other object that splits your epoch into batches and drives your training loop. You're trying to call scheduler more times per epoch than it is possible with batch size and epoch size values that you've supplied. times = samples_per_epoch // batch_size

[enzoampil]

does it have to be exactly equal to this number? Can it be less?

[enzoampil]

getting a new error:

---

TypeError Traceback (most recent call last)
in ()
1 training(model=model, epoch=20, eval_every=500,
2 loss_func=loss_function, optimizer=optimizer, train_iter=train_iter,
----> 3 val_iter=val_iter, scheduler=scheduler, warmup_epoch=3, early_stop=2)

in training(epoch, model, eval_every, loss_func, optimizer, train_iter, val_iter, scheduler, early_stop, warmup_epoch)
11
12 for e in range(epoch):
---> 13 scheduler.step()
14 #optimizer = exp_lr_scheduler(optimizer, e) #Added this line for learning rate decay
15 if e >= warmup_epoch:

in step(self)
270 self.last_epoch += 1
271 self.t_epoch += 1
--> 272 self._set_batch_size()
273 self.batch_step()
274

in _set_batch_size(self)
265 d, r = divmod(self.epoch_size, self.batch_size)
266 batches_in_epoch = d + 2 if r > 0 else d + 1
--> 267 self.batch_increment = iter(torch.linspace(0, 1, batches_in_epoch))
268
269 def step(self):

TypeError: 'generator' object is not callable

[mpyrozhok]

@enzoampil

does it have to be exactly equal to this number? Can it be less?

It will be a different scheduler then, probably. Could you elaborate on why do you need it to be less? And which parameters do you use while creating the scheduler? A brief explanation of your intentions and thoughts would also be helpful.