import ml_collections

def get_moon_to_spiral_default_configs():
  config = ml_collections.ConfigDict()
  # training
  config.training = training = ml_collections.ConfigDict()
  config.seed=42 #The utlimate answer of UNIVERSE!
  config.T = 1.0
  config.interval = 100
  config.train_method = 'alternate'
  config.t0 = 0
  config.problem_name = 'moon-to-spiral'
  config.num_itr = 250
  config.eval_itr = 200
  config.forward_net = 'toy'
  config.backward_net = 'toy'
  config.use_arange_t = True
  config.num_epoch = 1
  config.num_stage = 8
  config.train_bs_x = 1000
  config.sde_type = 'simple'
  # sampling
  config.samp_bs = 1000
  config.sigma_min = 0.1
  config.sigma_max = 2
  config.snapshot_freq = 2
  # optimization
#   config.optim = optim = ml_collections.ConfigDict()
  config.weight_decay = 0
  config.optimizer = 'AdamW'
  config.lr = 1e-3
  config.lr_gamma = 0.9

  model_configs=None
  return config, model_configs