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

Make callable through bicls_head

Browse Source
This commit is contained in:
Aayush
2019-01-07 00:02:39 +05:30
committed by GitHub
parent d0c15472d6
commit 8fe2891402
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastai_contrib/models.py
+1
View File
@@ -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()