from datasets import load_dataset

# 替换成你要用的任务名称
datasets_to_download = [
    "winogrande", 
    "hellaswag", 
    "arc", 
    "lambada_openai", 
    "piqa", 
    "trivia_qa"
]

for name in datasets_to_download:
    print(f"Downloading dataset: {name}")
    try:
        # 加载并缓存数据集（默认缓存路径 ~/.cache/huggingface/datasets）
        load_dataset(name)
    except Exception as e:
        print(f"Failed to download {name}: {e}")