Just taking a guess here…
Shouldn’t have to do this since it should default to the same device as the model. Maybe try: learn.dls.device = ‘cuda’ ?