소웨공1 드디어 lamda로 구운 나의 커스텀 모델 모델을 여러번 학습시켜봤지만 실제 서비스를 위해서 새로운 데이터를 넣어서 json 형태로 카테고리를 저장해본 경험은 처음이다! pytorch 모델을 pth확장자로 구워서 lamda를 이용해, AWS SageMaker에서 서비스 될 수 있도록 마무리할 예정이다. 새롭게 해본 경험이라 어렵기도 하지만 해결해서 기분이 많이 좋다! 이번주에 일본 학회도 다녀오면 대충 바쁜 일은 마무리 되지 않을까.. import torch from transformers import BertConfig, BertTokenizer, BertForSequenceClassification import json from mymodel import MyModel1 def load_model(model_path, number_of_lab.. 2023. 11. 21. 이전 1 다음 728x90
