Before_batch callback not called on last batch

fastai
1

I’m trying to build some variations on mixup, but the before_batch callback isn’t being executed on the last batch within each epoch.

Is there a way to change this so that the before_batch callback gets executed for every batch?

2

Set the downloader attribute train.drop_last = False to include the last partial batch in training.

e.g.

dls.train.drop_last = False