From 8fe28914026df63731baba99fcbc0655d7aacb04 Mon Sep 17 00:00:00 2001
From: Aayush <aayushyadav96@gmail.com>
Date: Mon, 7 Jan 2019 00:02:39 +0530
Subject: [PATCH] Make callable through bicls_head

---
 fastai_contrib/models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/fastai_contrib/models.py b/fastai_contrib/models.py
index 61a84c1..8cae758 100644
--- a/fastai_contrib/models.py
+++ b/fastai_contrib/models.py
@@ -272,6 +272,7 @@ def get_birnn_classifier(bptt:int, max_seq:int, n_class:int, vocab_sz:int, emb_s
     head = BiPoolingLinearClassifier
     if bicls_head == 'BiPoolingLinearClassifier': head = BiPoolingLinearClassifier
     elif bicls_head == 'AvgPoolingLinearClassifier': head = AvgPoolingLinearClassifier
+    elif bicls_head == 'BiAttentionPoolingClassifier': head = BiAttentionPoolingClassifier
 
     model = SequentialRNN(BiLMModel(fwd_rnn_enc, bwd_rnn_enc), head(layers, drops))
     model.reset()