Skip to content
Snippets Groups Projects
Commit 871a5bbd authored by Carel van Niekerk's avatar Carel van Niekerk :computer:
Browse files

Add setsumbt tracker to dst init

parent 413f971c
No related branches found
No related tags found
No related merge requests found
from convlab.dst.dst import DST
from convlab.dst.setsumbt import SetSUMBTTracker
from convlab.dst.setsumbt.tracker import SetSUMBTTracker
\ No newline at end of file
......@@ -78,11 +78,6 @@ class SetSUMBTTracker(DST):
self.device = torch.device('cuda') if USE_CUDA else torch.device('cpu')
# Value dict for value normalisation
path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
path = os.path.join(path, 'data/multiwoz/value_dict.json')
self.value_dict = json.load(open(path))
self.load_weights()
def load_weights(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment