I train resnext101_32x8d on the regression block and it finishes the training process but when I make a prediction I get this error
dls = ImageDataLoaders.from_df(train_hq, path=\Images (1)\\Images\\', suff='.jpeg', label_col='growth_stage', item_tfms=Resize(224), y_block=RegressionBlock, num_workers=0)
learn = cnn_learner(dls,models.resnext101_32x8d, metrics=rmse,cbs=WandbCallback()) learn.fine_tune(22,base_lr=0.00831763744354248)
ss = pd.read_csv('Test.csv')
tdl = learn.dls.test_dl(ss) # Make test dataloader preds = learn.tta(dl=tdl)# Test Time Augmentation
RuntimeError: Boolean value of Tensor with more than one value is ambiguous