wassname/multifit
1
0 Fork 0
mirror of https://github.com/wassname/multifit.git synced 2026-10-29 13:20:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove fastai warning

Browse Source
This commit is contained in:
NAUSICAA\Julian committed 2019-02-20 17:02:56 -03:00
1 parent c698c97772
commit 61ae344fda
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
results/time_benchmark/qrnn_benchmark.py
+2 -2
View File
@@ -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))