반응형
딥러닝 - 콜백을 이용하여 가장 좋은 모델과 로그 저장하기
이번 포스팅은 콜백을 이용하여 모델/로그를 저장하는 것을 중점으로 작성하였습니다. 딥러닝의 모델링 부분은 포함하지 않습니다. 모델/로그 저장 할 디렉토리 만들기 # 파이썬으로 디렉토리를 만드는 코드 작성 PROJECT_PATH = '.' # 기본 경로 설정, 현재 위치 model_type='mobilenetv2' # 사용중인 모델 # 모델과 로그기록 저장 경로 설정 # CHECKPOINT_PATH: ./checkpoints/mobilenetv2/mobilenetv2.h5 CHECKPOINT_PATH = PROJECT_PATH+'/checkpoints/'+model_type+'/'+model_type+'.h5' # LOGFILE_PATH: ./log/mobilenetv2/mobilenetv2-trainin..
- AI & ML/Machine Learning (Python)
- · 2022. 6. 19.
반응형