@dataloader('AMZ_computers')
def _load_amazon_computers():
    return tg.datasets.Amazon(root='/tmp/amazon', name='Computers')[0]
@dataloader('AMZ_computers')
def _load_amazon_computers():
    return tg.datasets.Amazon(root='/tmp/amazon', name='Computers')[0]