From 7fbf864d5f03753fde51228289c1af86da560597 Mon Sep 17 00:00:00 2001 From: Atif Ahmed Date: Mon, 23 Dec 2019 15:56:24 -0500 Subject: [PATCH] Avoiding duplicate downloads --- texar/torch/modules/classifiers/bert_classifier.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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)