ddpm #915
ddpm #915
Conversation
|
Thanks for your contribution! |
|
|
|
git 提交之前需要先用pre-commit进行检查，可参考https://zhuanlan.zhihu.com/p/590275006等，安装pre-commit |
| @@ -0,0 +1,64 @@ | |||
| mode: train | |||
There was a problem hiding this comment.
config文件格式请参考其他案例（如aneurysm）修改，增加hydra部分，剩余参数分为MODEL/TRAIN/EVAL和其他，并精简删除没有用到的参数
| resume_training: false | ||
| data: | ||
| dataset: "kolmogorov flow" | ||
| data_dir: "/home/aistudio/data/data264003/train_data.npy" |
There was a problem hiding this comment.
修改路径为ddpm下的相对路径，之后yaml文件中的其他路径也是
| @@ -0,0 +1,50 @@ | |||
| log_dir: "./experiments/kmflow_re1000_rs256_ddim_recons_conditional_log" | |||
| @@ -0,0 +1,201 @@ | |||
| import paddle | |||
There was a problem hiding this comment.
请删除函数中的注释（除了特别需要解释的部分），复杂函数开头写上hint，注释需要使用英文、半角，修改后请使用pre-commit检查代码格式
| dx_func = kwargs.get("dx_func", None) | ||
| clamp_func = kwargs.get("clamp_func", None) | ||
| cache = kwargs.get("cache", False) | ||
| logger = kwargs.get("logger", None) |
There was a problem hiding this comment.
请参考paddlescience格式修改：
- 此处是否可以删除某些变量的设置，如logger
- kwargs.get试图获取“ dx_func”对应的变量，如果没有就设置dx_func=None，请参考其他example将这种写法修改为def ddim_steps(dx_func:xxtype(如int)=None)
| def transform_out(outputs): | ||
| loss = (outputs[1] - outputs[0]).square().sum(axis=(1, 2, 3)).mean(axis=0) | ||
|
|
||
| return {"loss": loss} |
| Returns: | ||
| paddle.Tensor: Loss value. | ||
| """ | ||
| return result_dict["loss"] |
| return x | ||
|
|
||
|
|
||
| def voriticity_residual_train(w, re=1000.0, dt=1 / 32): |
There was a problem hiding this comment.
这部分和functions中的有重复吗？如果有的话需要提取和整理，尽量避免重复代码
|
|
||
|
|
||
| def get_beta_schedule_train( | ||
| beta_schedule, *, beta_start, beta_end, num_diffusion_timesteps |
| return betas | ||
|
|
||
|
|
||
| class EMAHelper(object): |
PR types
new feature
PR changes
new ddpm.py in ppsci.arch;new ddpm folder in example folder.
Describe