8 lines
303 B
Python
8 lines
303 B
Python
|
# from transformers.models.bert.modeling_bert import BertForMaskedLM
|
||
|
|
from models.huggingface_relformer import BertForMaskedLM
|
||
|
|
class BertKGC(BertForMaskedLM):
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def add_to_argparse(parser):
|
||
|
|
parser.add_argument("--pretrain", type=int, default=0, help="")
|
||
|
|
return parser
|