classifier_activation

dmlc · Jul 29, 2020 · 9d60cda · 9d60cda
1 parent 510d991
commit 9d60cda
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/gluonnlp/models/bart.py b/src/gluonnlp/models/bart.py
@@ -289,6 +289,7 @@ def from_cfg(cls, cfg, dtype=None,
                    tie_weights=cfg.MODEL.tie_weights,
                    data_norm=cfg.MODEL.data_norm,
                    use_pooler=use_pooler,
+                   classifier_activation=classifier_activation,
                    attention_dropout=cfg.MODEL.attention_dropout,
                    activation_dropout=cfg.MODEL.activation_dropout,
                    dropout=cfg.MODEL.dropout,