Hi @sgugger. Tried to update to the last version of fastai2 (v 0.0.8) and now receiving an error when using text_classifier_learner:
learn = text_classifier_learner(dbunch_fwd,
AWD_LSTM,
seq_len=72,
pretrained=False,
config=config,
metrics=[accuracy],
path=path,
drop_mult=0.7,
loss_func=CrossEntropyLossFlat()
).to_fp16()
error:
---------------------------------------------------------------------------
NameError Traceback (most recent call last)
<ipython-input-30-a900cdbe143c> in <module>
7 path=path,
8 drop_mult=0.7,
----> 9 loss_func=CrossEntropyLossFlat()
10 ).to_fp16()
11
/media/hdd3tb/data/fastai2/fastai2/text/learner.py in text_classifier_learner(dls, arch, seq_len, config, pretrained, drop_mult, n_out, lin_ftrs, ps, max_len, **kwargs)
173 drop_mult=drop_mult, lin_ftrs=lin_ftrs, ps=ps, max_len=max_len)
174 meta = _model_meta[arch]
--> 175 learn = TextLearner(dls, model, splitter=meta['split_clas'], **kwargs)
176 if pretrained:
177 if 'url' not in meta:
/media/hdd3tb/data/fastai2/fastai2/text/learner.py in __init__(self, model, dls, loss_func, alpha, beta, moms, **kwargs)
51 def __init__(self, model, dls, loss_func, alpha=2., beta=1., moms=(0.8,0.7,0.8), **kwargs):
52 super().__init__(model, dls, loss_func, moms=moms, **kwargs)
---> 53 self.add_cbs([ModelReseter(), RNNTrainer(alpha=alpha, beta=beta)])
54
55 def save_encoder(self, file):
NameError: name 'RNNTrainer' is not defined