def from_coco_128(
    batch_size: int = 4,
    num_workers: int = 0,
    **data_module_kwargs,
) -> StyleTransferData:
    """Downloads and loads the COCO 128 data set."""
    download_data("https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.3.0/coco128.zip", "data/")
    return StyleTransferData.from_folders(
        train_folder="data/coco128/images/train2017/",
        batch_size=batch_size,
        num_workers=num_workers,
        **data_module_kwargs,
    )
Downloads and loads the COCO 128 data set.
src/flash/image/style_transfer/cli.py