Skip to content

Commit 9a313c9

Browse files
committed
🔧 Increase DDPM size and reduce dropout in ddpm
1 parent a5e96bb commit 9a313c9

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

icenet_mp/config/model/04_ddpm.yaml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,3 +2,6 @@ _target_: icenet_mp.models.ddpm.DDPM
22

33
# Run DDPM model with default settings
44
name: ddpm
5+
dropout_rate: 0.05 # reduce interference in denoising step
6+
start_out_channels: 64 # increase model size
7+
timesteps: 250 # speed up training

0 commit comments

Comments
 (0)