SungaHwang / Model-Compression

A Lightweight Method for Large-scale Deep Learning Models
0 stars 0 forks source link

[~4/14] 추가할 것 #2

Closed SungaHwang closed 7 months ago

SungaHwang commented 7 months ago
  1. relations 데이터를 인풋으로 가져와서 적용할 수 있도록 MST 알고리즘 (Floyld, Kruskal, Prim) 3가지 구현해서 라이브러리 형태로 불러와서 쓸 수 있도록 하기
  2. relations 들로 각 MST 알고리즘 적용해서 불필요한 가중치 파라미터 제거한 모델 구축 후, 추론 성능 비교 및 확인
SungaHwang commented 7 months ago

1번: Floyd 제외하고 해결 2번: MST 알고리즘 적용해서 불필요한 가중치 파라미터를 제거하는 과정까지는 해결 - 수정된 가중치를 기존 모델에 집어 넣는 과정에서 막힘

SungaHwang commented 7 months ago

2번 해결. 그러나 정확도가 오히려 상승을 해버려서 코드 다시 검토 필요.

SungaHwang commented 7 months ago

Floyd 해결