From e28bc3ddb3bb9c322803e7c8204fb9246dee72ee Mon Sep 17 00:00:00 2001 From: ca61zope <chang.peng@fau.de> Date: Thu, 15 May 2025 19:01:43 +0200 Subject: [PATCH] Update --- README.md | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 5ce765b4f..443286f8c 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 -- GitLab