From 9d60cdab9cfcb5705e779a97749959bdb1d5a23e Mon Sep 17 00:00:00 2001 From: ZheyuYe Date: Thu, 30 Jul 2020 03:17:28 +0800 Subject: [PATCH] classifier_activation --- src/gluonnlp/models/bart.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/gluonnlp/models/bart.py b/src/gluonnlp/models/bart.py index fe61fab37e..6b011a7566 100644 --- 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,