Avoiding duplicate downloads

asyml · Dec 26, 2019 · 7d56607 · 7d56607
1 parent 824d947
commit 7d56607
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)