Upload app.yaml

config/app.yaml

@@ -0,0 +1,109 @@
+device: "NVIDIA GeForce RTX 2080 Ti"
+
+host: 127.0.0.1
+port: 7860
+
+is_push_to_public: false
+save-path: save/stack/outputs.jsonl
+page-size: 2
+
+base:
+  name: "OpenSLUv1"
+  train: false
+  test: false
+  device: cuda
+  ckpt_path: null
+  seed: 42
+  best_key: EMA
+  epoch_num: 300
+  batch_size: 16
+  eval_by_epoch: true
+  model_dir: save/stack
+  template: application.html
+accelerator:
+  use_accelerator: false
+
+dataset:
+  dataset_name: atis
+
+metric:
+  - intent_acc
+  - slot_f1
+  - EMA
+
+tokenizer:
+    _tokenizer_name_: word_tokenizer
+    _padding_side_: right
+    _align_mode_: fast
+    _to_lower_case_: true
+    add_special_tokens: false
+    max_length: 512
+
+optimizer:
+  _model_target_: torch.optim.Adam
+  _model_partial_: true
+  lr: 0.001
+  weight_decay: 1e-6
+
+scheduler:
+  _model_target_: transformers.get_scheduler
+  _model_partial_: true
+  name : "linear"
+  num_warmup_steps: 0
+
+model:
+  _model_target_: model.OpenSLUModel
+
+  encoder:
+    _model_target_: model.encoder.AutoEncoder
+    encoder_name: self-attention-lstm
+
+    embedding:
+      embedding_dim: 256
+      dropout_rate: 0.55
+
+    lstm:
+      layer_num: 1
+      bidirectional: true
+      output_dim: 256
+      dropout_rate: 0.5
+
+    attention:
+      hidden_dim: 1024
+      output_dim: 128
+      dropout_rate: 0.6
+
+    return_with_input: true
+    return_sentence_level_hidden: false
+
+  decoder:
+    _model_target_: model.decoder.StackPropagationDecoder
+    interaction:
+      _model_target_: model.decoder.interaction.StackInteraction
+      differentiable: false
+
+    intent_classifier:
+      _model_target_: model.decoder.classifier.AutoregressiveLSTMClassifier
+      layer_num: 1
+      bidirectional: false
+      force_ratio: 0.9
+      hidden_dim: 64
+      embedding_dim: 8
+      ignore_index: -100
+      dropout_rate: 0.5
+      mode: "token-level-intent"
+      use_multi: false
+      return_sentence_level: true
+
+    slot_classifier:
+      _model_target_: model.decoder.classifier.AutoregressiveLSTMClassifier
+      layer_num: 1
+      bidirectional: false
+      force_ratio: 0.9
+      hidden_dim: 64
+      embedding_dim: 32
+      ignore_index: -100
+      dropout_rate: 0.55
+      mode: "slot"
+      use_multi: false
+      return_sentence_level: false