From 591393b96c21a6a68d6da166bc391237dee9102e Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Sat, 8 Dec 2018 23:48:02 +0100 Subject: [PATCH] Change AvgPooling to BiPooling as a default --- fastai_contrib/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastai_contrib/models.py b/fastai_contrib/models.py index 57ccc47..0c328a7 100644 --- a/fastai_contrib/models.py +++ b/fastai_contrib/models.py @@ -145,7 +145,7 @@ def get_birnn_classifier(bptt:int, max_seq:int, n_class:int, vocab_sz:int, emb_s bwd_rnn_enc = MultiBatchRNNCore(bptt, max_seq, vocab_sz, emb_sz, n_hid, n_layers, pad_token=pad_token, bidir=bidir, qrnn=qrnn, hidden_p=hidden_p, input_p=input_p, embed_p=embed_p, weight_p=weight_p) - model = SequentialRNN(BiLMModel(fwd_rnn_enc, bwd_rnn_enc), AvgPoolingLinearClassifier(layers, drops)) + model = SequentialRNN(BiLMModel(fwd_rnn_enc, bwd_rnn_enc), BiPoolingLinearClassifier(layers, drops)) model.reset() return model