A small note because I couldn't find this easily (even though it is documented). To train a regression task in fast.ai you need to specify the y_block
as RegressionBlock
.
For example with ImageDataLoaders:
dls = ImageDataLoaders.from_df(df, path, bs=16, y_block=RegressionBlock)