Might need tags by datasets used, i.e. mnist is image classification, COCO is object recognition, etc.