model:
  descriptor:
    type: loc_frame
    sel_a:
    - 16
    - 32
    sel_r:
    - 30
    - 60
    rcut: 6.0
    axis_rule: [0, 1, 0, 0, 1, 1, 0, 0, 0, 0, 1, 0]
  fitting_net:
    neuron:
    - 240
    - 120
    - 60
    - 30
    - 10
    resnet_dt: true
    seed: 1
learning_rate:
  type: exp
  decay_steps: 5000
  decay_rate: 0.95
  start_lr: 0.001
loss:
  start_pref_e: 0.02
  limit_pref_e: 8
  start_pref_f: 1000
  limit_pref_f: 1
  start_pref_v: 0
  limit_pref_v: 0
training:
  systems:
  - ../data/
  set_prefix: set
  stop_batch: 1000000
  batch_size:
  - 4
  seed: 1
  disp_file: lcurve.out
  disp_freq: 100
  numb_test: 10
  save_freq: 1000
  save_ckpt: model.ckpt
  disp_training: true
  time_training: true
