adding save configure

SKTBrain · Aug 22, 2019 · f1d0237 · f1d0237
1 parent 402ecd6
commit f1d0237
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/code/main-multislot.py b/code/main-multislot.py
@@ -625,6 +625,7 @@ def main():
     if args.fp16:
         model.half()
     model.to(device)
+    save_configure(args, num_labels, processor.ontology)
 
     ## Get slot-value embeddings
     label_token_ids, label_len = [], []
@@ -1014,6 +1015,23 @@ def _eval_acc(_pred_slot, _labels):
 
     return accuracies
 
+def save_configure(args, num_labels, ontology):
+    with open(os.path.join(args.output_dir, "config.json"),'w') as outfile:
+        data = { "hidden_dim": args.hidden_dim,
+                "num_rnn_layers": args.num_rnn_layers,
+                "zero_init_rnn": args.zero_init_rnn,
+                "max_seq_length": args.max_seq_length,
+                "max_label_length": args.max_label_length,
+                "num_labels": num_labels,
+                "attn_head": args.attn_head,
+                 "distance_metric": args.distance_metric,
+                 "fix_utterance_encoder": args.fix_utterance_encoder,
+                 "task_name": args.task_name,
+                 "bert_dir": args.bert_dir,
+                 "bert_model": args.bert_model,
+                 "do_lower_case": args.do_lower_case,
+                 "ontology": ontology}
+        json.dump(data, outfile, indent=4)
 
 if __name__ == "__main__":
     main()