diff --git a/texar/torch/modules/classifiers/bert_classifier.py b/texar/torch/modules/classifiers/bert_classifier.py index a0dc2f8e8..9cc0a1b6f 100644 --- a/texar/torch/modules/classifiers/bert_classifier.py +++ b/texar/torch/modules/classifiers/bert_classifier.py @@ -76,9 +76,10 @@ def __init__(self, # Create the underlying encoder encoder_hparams = dict_fetch(hparams, self._ENCODER_CLASS.default_hparams()) + encoder_hparams['pretrained_model_name'] = None self._encoder = self._ENCODER_CLASS( - pretrained_model_name=pretrained_model_name, + pretrained_model_name=None, cache_dir=cache_dir, hparams=encoder_hparams)