From 61ae344fda4ec7ddf6cd1e34b163459194ab686e Mon Sep 17 00:00:00 2001 From: "NAUSICAA\\Julian" Date: Wed, 20 Feb 2019 17:02:56 -0300 Subject: [PATCH] Remove fastai warning --- results/time_benchmark/qrnn_benchmark.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/results/time_benchmark/qrnn_benchmark.py b/results/time_benchmark/qrnn_benchmark.py index 154ec9a..f6dab86 100644 --- a/results/time_benchmark/qrnn_benchmark.py +++ b/results/time_benchmark/qrnn_benchmark.py @@ -16,8 +16,8 @@ for mode in ['train', 'test']: for file in glob.glob(pattern): shutil.copy(file, tgt_path) -data_lm = TextLMDataBunch.from_folder(path) -data_clas = TextClasDataBunch.from_folder(path, bs=32, vocab=data_lm.train_ds.vocab) +data_lm = TextLMDataBunch.from_folder(path, valid='test') +data_clas = TextClasDataBunch.from_folder(path, bs=32, vocab=data_lm.train_ds.vocab, valid='test') print('Vocab size', len(data_lm.train_ds.vocab.itos))