fix the debug

2019-09-17 18:44:53 +02:00 · 2019-09-17 18:44:53 +02:00 · c8a548d375
parent 8d3775a7d6
commit c8a548d375
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -62,7 +62,7 @@ def setup_loader(ap, is_val=False, verbose=False):
        dataset = MyDataset(
            c.r,
            c.text_cleaner,
-            meta_data=meta_data_eval if is_val else meta_data_train[:64],
+            meta_data=meta_data_eval if is_val else meta_data_train,
            ap=ap,
            batch_group_size=0 if is_val else c.batch_group_size * c.batch_size,
            min_seq_len=c.min_seq_len,