File dataset_benchmark.py

code/dataset_benchmark.py:None–None

import os
import shutil
from typing import Callable, List, Optional, Tuple, Union
import torch