https://github.com/lonepatient/albert_pytorch/blob/master/readme_zh.md
Primeiro lugar albert_zh neste repositório sob seu projeto
from albert_zh import ...
AlbertConfig
AlbertTokenizer
AlbertModel
AlbertForMaskedLM
AlbertForQuestionAnswering
AlbertForSequenceClassification
https://huggingface.co/transformers/v2.3.0/model_doc/albert.html
usage_example.pyOu consulte o exemplo de uso real de p208p2002/taipei-qa-bert
Essa nomeação de repositório não cumpre a convenção de nomeação do módulo Python, e o próprio Jupyter não tem um bom suporte para módulos personalizados. Consulte o exemplo da solução abaixo. Considere empurrar Pypi mais tarde
# 此段code僅適用於jupyter、colab
!git clone https://github.com/p208p2002/albert-zh-for-pytorch-transformers.git albert
import sys
sys.path.append('.')
from albert.albert_zh import AlbertConfig, AlbertTokenizer, AlbertForSequenceClassification
Verifique se a classe modelo e a configuração do modelo são introduzidas por Albert_ZH, não Transformers
Lonepaciente/Albert_pytorch#35
config.json adiciona "share_type":"all"
Use log() em vez de print() e execute blockPrint() uma vez no início do programa
import os , sys
def log ( * logs ):
enablePrint ()
print ( * logs )
blockPrint ()
# Disable
def blockPrint ():
sys . stdout = open ( os . devnull , 'w' )
# Restore
def enablePrint ():
sys . stdout = sys . __stdout__