Skip to content

Commit

Permalink
add missing values to default config for det models
Browse files Browse the repository at this point in the history
NCSN++ needs things like sigma_max set
  • Loading branch information
henryaddison committed Aug 5, 2024
1 parent e42ccd2 commit a6531c8
Showing 1 changed file with 6 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -64,9 +64,15 @@ def get_default_configs():

# model
config.model = model = ml_collections.ConfigDict()
model.sigma_min = 0.01
model.sigma_max = 50
model.beta_min = 0.1
model.beta_max = 20.
model.loc_spec_channels = 0
model.num_scales = 0
model.ema_rate = 0.9999
model.dropout = 0.1
model.embedding_type = 'fourier'

# optimization
config.optim = optim = ml_collections.ConfigDict()
Expand Down

0 comments on commit a6531c8

Please sign in to comment.