Searched refs:BERTTagger (Results 1 – 3 of 3) sorted by relevance
25 class BERTTagger(Block): class43 super(BERTTagger, self).__init__(prefix=prefix, params=params)
28 from model import BERTTagger88 net = BERTTagger(bert_model, dataset.num_tag_types, train_config.dropout_prob)
31 from model import BERTTagger, attach_prediction104 net = BERTTagger(bert_model, dataset.num_tag_types, config.dropout_prob)