make docstring in run batch sweep more generic
make docstring in run batch sweep more generic

chendaniely/multidisciplinary-diffusion-model-experiments,chendaniely/multidisciplinary-diffusion-model-experiments,chendaniely/multidisciplinary-diffusion-model-experiments