python.pytorch - k821209/pipelines GitHub Wiki

huggingface model등 저장

# 저장할때
best_model.save_pretrained('./saved_model/')
tokenizer.save_pretrained('./saved_model/')

# 불러울때
tokenizer = BertTokenizer.from_pretrained('./saved_model/')