does data_loader need specific settings or just change name of   IMNET to COCO
elif args.data_set == ‘COCO’:
if not args.use_mcloader:
root = os.path.join(args.data_path, ‘train’ if is_train else ‘val’)
dataset = datasets.ImageFolder(root, transform=transform)
else:
dataset = ClassificationDataset(
‘train’ if is_train else ‘val’,
pipeline=transform
)
nb_classes = 80