異端開発室

RobertaモデルのONNX変換と文脈マスク予測 モデルのONNX形式への変換 モデルはHugging Faceから入手可能です。 def setup_nlp_components(): tokenizer = BertTokenizer.from_pretrained("../model/chinese_roberta_L-2_H-512") model = BertForMaskedLM.from_pretrained("../model/chinese_roberta_L-2_H-512") return model, tokenizer def save_as_onnx ...