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.