Skip to content

[FEAT] transformer weights_path 로딩 로직 개선 필요 #73

@coderabbitai

Description

@coderabbitai

설명

transformer 추론 시 학습된 가중치를 로드하는 로직에 문제가 있습니다.

현재 문제점

  • AI/transformer/main.py에서 하드코딩된 절대경로(/transformer/weights/inital.weights.h5)만 확인
  • 파일명 오타: initalinitial로 수정 필요
  • config["transformer"]["weights_path"] 설정을 활용하지 않음
  • 학습 파이프라인이 저장하는 경로(transformer/weights/initial.weights.h5)와 불일치
  • 결과적으로 학습된 가중치가 존재해도 로드되지 않고 랜덤 초기화된 모델로 추론 실행

개선 방향

  1. config에서 weights_path를 우선적으로 읽어 절대경로로 정규화
  2. 미지정 시 현재 모듈 기준 상대경로 weights/initial.weights.h5 확인
  3. 파일명 철자 수정: initalinitial

참고

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions