diff --git a/README.md b/README.md index 5ce765b4f1c0c4d8e845d1ebff8cc0cb7a4002ab..443286f8c82dd787dee62dafde4c8d29a14e26ff 100644 --- a/README.md +++ b/README.md @@ -7,17 +7,18 @@ pip install -r requirements.txt ``` ### Data -Dataset here is Cylinder2D. +Dataset here is Cylinder2D with 3 channels. +I will finish a script to download the dataset I used in the future. -### Traning +### Training All .py are in the folder DDPM. -Use ddpm_conditional.py to train DDPM with conditional information through CFG. +Use DDPM/ddpm_conditional.py to train DDPM with conditional information through CFG. -Use interpolate.py to train Interpolator. +Use DDPM/interpolate.py to train Interpolator. -Use forecast.py to train Forecaster. +Use DDPM/forecast.py to train Forecaster. ### Result