It seems as if some data tensors are float64 tensors, while the default float32 is expected.
What does ImageDataBunch.from_name_re do?
Assuming it’s part of the data loading pipeline, are you able to somehow define the desired data type?