Python BertModel.__init__の例

コード例 #1

ファイル: hf_models.py プロジェクト: facebookresearch/DPR

 def __init__(self, config, project_dim: int = 0):
     BertModel.__init__(self, config)
     assert config.hidden_size > 0, "Encoder hidden_size can't be zero"
     self.encode_proj = nn.Linear(config.hidden_size,
                                  project_dim) if project_dim != 0 else None
     self.init_weights()

コード例 #2

ファイル: models.py プロジェクト: microsoft/SEED-Encoder

 def __init__(self, config):
     BertModel.__init__(self, config)
     assert config.hidden_size > 0, 'Encoder hidden_size can\'t be zero'
     self.init_weights()