config.data = Config({
"vocab_size": vocab_size,
"train_set": {
"type": "utterance_with_cands_from_negative_sampling",
"kwargs": {
"train": True,
"num_negs": 19,
"num_turns": 3
},
"loader": {
"batch_size": 30,
"shuffle": True,
"num_workers": 8,
}
},
"test_set": {
"type": "utterance_with_cands",
"kwargs": {
"train": False,
"num_turns": 3
},
"loader": {
"batch_size": 1,
"shuffle": False,
}
},
"corpus_set": {
"type": "utterance",
"kwargs": {
"train": True
},
"loader": {
"batch_size": 30,
"shuffle": False,
"num_workers": 8
}
}
})
