@dataloader('AMZ_photo')
def _load_amazon_photos():
    return tg.datasets.Amazon(root='/tmp/amazon', name='photo')[0]
@dataloader('AMZ_photo')
def _load_amazon_photos():
    return tg.datasets.Amazon(root='/tmp/amazon', name='photo')[0]