Paper on Imbalanced classes

[ashwinakannan (Ashwin)]

Hi,

@aipitch

Does it work with the current version of Fast.ai (1.0) and PyTorch (1.0)

I don’t think I’ve updated to the current version of fastai… But its fastai 1.0 & PyTorch 1.0

You have to make changes to the dataloader class. Labels in fastai is stored in LabelList… you’ll have to do something like:

dataset_type = type(dataset)

if dataset_type is fastai.data_block.LabelList:
   return dataset[idx][1].__int__()