Skip to content
Snippets Groups Projects
Unverified Commit 350f36c4 authored by Carel van Niekerk's avatar Carel van Niekerk Committed by GitHub
Browse files

Setsumbt updates

parent ffb3dc42
No related branches found
No related tags found
No related merge requests found
Expand all Show whitespace changes
Inline Side-by-side
convlab/dst/setsumbt/utils/ensemble.py 0 → 100644
+ 116
− 0
View file @ 350f36c4
This diff is collapsed.
convlab/policy/ppo/configs/RuleUser-TemplateNLG-SetSUMBT-VectorUncertainty.json
+ 2
− 2
View file @ 350f36c4
......@@ -26,9 +26,9 @@
"nlu_sys": {},
"dst_sys": {
"setsumbt-mul": {
"class_path": "convlab.dst.setsumbt.SetSUMBTTracker",
"class_path": "convlab.dst.setsumbt.tracker.SetSUMBTTracker",
"ini_params": {
"model_path": "https://huggingface.co/ConvLab/setsumbt-dst_nlu-multiwoz21-EnD2/resolve/main/SetSUMBT-nlu-multiwoz21-roberta-gru-cosine-distribution_distillation-Seed0.zip",
"model_path": "ConvLab/setsumbt-dst_nlu-multiwoz21-EnD2",
"return_confidence_scores": true,
"return_belief_state_mutual_info": true
}
......
convlab/policy/ppo/configs/RuleUser-TemplateNLG-SetSUMBT.json
+ 2
− 2
View file @ 350f36c4
......@@ -24,9 +24,9 @@
"nlu_sys": {},
"dst_sys": {
"setsumbt-mul": {
"class_path": "convlab.dst.setsumbt.SetSUMBTTracker",
"class_path": "convlab.dst.setsumbt.tracker.SetSUMBTTracker",
"ini_params": {
"model_path": "https://huggingface.co/ConvLab/setsumbt-dst_nlu-multiwoz21-EnD2/resolve/main/SetSUMBT-nlu-multiwoz21-roberta-gru-cosine-distribution_distillation-Seed0.zip"
"model_name_or_path": "ConvLab/setsumbt-dst_nlu-multiwoz21-EnD2"
}
}
},
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment