RobertaTokenizer,RobertaForMaskedLM

RobertaTokenizer,RobertaForMaskedLM

from transformers import RobertaTokenizer,RobertaForMaskedLM
tokenizer = RobertaTokenizer.from_pretrained('microsoft/graphcodebert-base')
model = RobertaForMaskedLM.from_pretrained('microsoft/graphcodebert-base')
input_ids = tokenizer(['Language model is what I need.','I love China'],padding=True,return_tensors='pt')
print(input_ids)
#out = model(**input_ids)
# print(out)

RobertaTokenizer,RobertaForMaskedLM_第1张图片

你可能感兴趣的:(Python,Pytorch,python,深度学习)