Skip to content
Snippets Groups Projects
Commit e28bc3dd authored by Chang Peng's avatar Chang Peng
Browse files

Update

parent 92a04da2
No related branches found
No related tags found
No related merge requests found
...@@ -7,17 +7,18 @@ pip install -r requirements.txt ...@@ -7,17 +7,18 @@ pip install -r requirements.txt
``` ```
### Data ### 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. 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 ### Result
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment