search

openai / improved-diffusion

Release for Improved Denoising Diffusion Probabilistic Models
MIT License
3.22k stars 483 forks source link

Problem with training one's own dataset #122

Open liuyisu opened 9 months ago

liuyisu commented 9 months ago

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

ali-alhwayzee commented 5 months ago

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 commented 1 month ago

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