movie_script persona maker

gorkemgoknar · Dec 18, 2020 · 35a50f5 · 35a50f5
1 parent af09ca6
commit 35a50f5
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 0 deletions.
diff --git a/__pycache__/utils.cpython-36.pyc b/__pycache__/utils.cpython-36.pyc
diff --git a/train.py b/train.py
@@ -75,6 +75,7 @@ def get_data_loaders(args, tokenizer):
     logger.info("Build inputs and labels")
     datasets = {"train": defaultdict(list), "valid": defaultdict(list)}
     for dataset_name, dataset in personachat.items():
+        print(dataset[0]["utterances"][0])
         num_candidates = len(dataset[0]["utterances"][0]["candidates"])
         if args.num_candidates > 0 and dataset_name == 'train':
             num_candidates = min(args.num_candidates, num_candidates)