Problem with training one's own dataset

[liuyisu]

What should we do if this problem occurs ModuleNotFoundError: No module named 'improved_diffusion'

[ali-alhwayzee]

There are two possibilities, according to what happened to me:

1. The first is to ensure that MPI is installed https://www.microsoft.com/en-us/download/details.aspx?id=105289
2. The second is that the path is inside the "improved_diffusion" file After setting the parameters, use the following command inside the terminal in the path above python scripts/image_train.py --data_dir path/to/images $MODEL_FLAGS $DIFFUSION_FLAGS $TRAIN_FLAGS Make sure to check the image file path "--data_dir path/to/images"

[PandragonXIII]

Just add the working dir to python environment variable as the following: export PYTHONPATH=$PYTHONPATH:/path/to/guided-diffusion hopefully this will help.