custom_model_t.yaml 134 B

12345678
  1. _base_: ../levircd.yaml
  2. save_dir: ./exp/levircd/ablation/att_t/
  3. model: !Node
  4. type: CustomTrainer
  5. args:
  6. att_types: t