Python BERT2BERTConfigの例

コード例 #1

def get_model_params(model: Optional[Text] = "bert2bert",
                     config_class=None) -> params_dict.ParamsDict:
  """Helper function to convert config file to ParamsDict."""
  if model == "bert2bert":
    return configs.BERT2BERTConfig()
  elif model == "nhnet":
    return configs.NHNetConfig()
  elif config_class:
    return config_class()
  else:
    raise KeyError("The model type is not defined: %s" % model)

コード例 #2

ファイル: models_test.py プロジェクト: mkonaite2/models2

 def setUp(self):
     super(NHNetTest, self).setUp()
     self._nhnet_config = configs.NHNetConfig()
     self._nhnet_config.override(utils.get_test_params().as_dict())
     self._bert2bert_config = configs.BERT2BERTConfig()
     self._bert2bert_config.override(utils.get_test_params().as_dict())

コード例 #3

ファイル: configs_test.py プロジェクト: avs11011/TF-sMLModelGarden

 def test_configs(self):
     cfg = configs.BERT2BERTConfig()
     cfg.validate()
     self.assertEqual(cfg.as_dict(), BERT2BERT_CONFIG)