wassname/multifit
1
0
Fork 0
mirror of https://github.com/wassname/multifit.git synced 2026-09-09 11:27:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated method to new fastai v1 version

Browse Source
This commit is contained in:
Sebastian
2018-11-15 17:51:44 +00:00
parent 94fa1a3ecb
commit c9c827c420
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/train_clas.py
+5 -4
View File
@@ -95,12 +95,13 @@ def new_train_clas(data_dir, lang='en', cuda_id=0, pretrain_name='wt-103', model
print(f'Train size: {len(ids[TRN])}. Valid size: {len(ids[VAL])}. '
f'Test size: {len(ids[TST])}.')
data_lm = TextLMDataBunch.from_ids(path=tmp_dir, vocab=vocab, trn_ids=ids[TRN],
val_ids=ids[VAL], bs=bs, bptt=bptt)
data_lm = TextLMDataBunch.from_ids(path=tmp_dir, vocab=vocab, train_ids=ids[TRN],
valid_ids=ids[VAL], bs=bs, bptt=bptt)
# TODO TextClasDataBunch allows tst_ids as input, but not tst_lbls?
data_clas = TextClasDataBunch.from_ids(
path=tmp_dir, vocab=vocab, trn_ids=ids[TRN], val_ids=ids[VAL],
trn_lbls=lbls[TRN], val_lbls=lbls[VAL], bs=bs)
path=tmp_dir, vocab=vocab, train_ids=ids[TRN], valid_ids=ids[VAL],
test_ids=ids[TST], train_lbls=lbls[TRN], valid_lbls=lbls[VAL],
test_lbls=lbls[TST], bs=bs)
if qrnn:
emb_sz, nh, nl = 400, 1550, 3