config.demo.yaml [24:31]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
optim:
  _target_: flax.optim.Adam
  learning_rate: 1e-3
  beta1: 0.9
  beta2: 0.999

flow:
    _target_: flows.SequentialFlow
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



config.yaml [43:50]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
optim:
  _target_: flax.optim.Adam
  learning_rate: 1e-3
  beta1: 0.9
  beta2: 0.999

flow:
    _target_: flows.SequentialFlow
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



