RuntimeError: number of categories cannot exceed 2^24

[LeeBinshan]

train_loader = DataLoader(dataset=train_data, 
                          sampler=ImbalancedDatasetSampler(train_data),  # sampler for imbalanced dataset
                          batch_size=args.batch_size,
                          shuffle=False,
                          num_workers=args.workers,
                          drop_last=True)