Can't make MixUp work

AllBecomesGood · July 13, 2022, 10:06am

At the end of Chapter 14 it is suggested to try training the ResNet with MixUp.

l = Learner(dls, rn, loss_func=nn.CrossEntropyLoss(), metrics=accuracy, cbs=MixUp()).to_fp16()
l.fit_one_cycle(100, 3e-3)

Afaik using the library all that is needed is to add the cbs=MixUp() part. However, this error occurs:

RuntimeError: Exception occured in MixUp when calling event before_batch:
expected dtype long int for weight but got dtype float