deepar.yaml 302 B

12345678910111213
  1. # SPDX-License-Identifier: Apache-2.0
  2. _target_: models.deepar.DeepAR
  3. config:
  4. model_type: autoregressive
  5. num_layers: 3
  6. hidden_size: 40
  7. use_embedding: true
  8. embedding_dim: 20
  9. dropout: 0.1
  10. quantiles: [0.5, 0.9]
  11. defaults:
  12. - _self_
  13. - /trainer@_global_/trainer: ctltrainer