-
Notifications
You must be signed in to change notification settings - Fork 2
Closed
Description
설명
transformer 추론 시 학습된 가중치를 로드하는 로직에 문제가 있습니다.
현재 문제점
AI/transformer/main.py에서 하드코딩된 절대경로(/transformer/weights/inital.weights.h5)만 확인- 파일명 오타:
inital→initial로 수정 필요 config["transformer"]["weights_path"]설정을 활용하지 않음- 학습 파이프라인이 저장하는 경로(
transformer/weights/initial.weights.h5)와 불일치 - 결과적으로 학습된 가중치가 존재해도 로드되지 않고 랜덤 초기화된 모델로 추론 실행
개선 방향
- config에서 weights_path를 우선적으로 읽어 절대경로로 정규화
- 미지정 시 현재 모듈 기준 상대경로
weights/initial.weights.h5확인 - 파일명 철자 수정:
inital→initial
참고
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels