访问持久存储 Accessing persistent storage

import fal
from pathlib import Path

DATA_DIR = Path("/data/mnist")

@fal.function(
    "virtualenv",
    requirements=["torch>=2.0.0", "torchvision"],
    machine_type="M",
)
def train_fashion_model():
    import torch
    from torchvision import datasets

    already_present = DATA_DIR.exists()
    if already_present:
        print("Test data is already downloaded, skipping download!")

    test_data = datasets.FashionMNIST(
        root=DATA_DIR,
        train=False,
        download=not already_present,
    )
    ...

if __name__ == "__main__":
    train_fashion_model()