Hns hanul feature resnet classifier - Githubissues

ML-TANGO / TANGO

public repo for TANGO (Target Aware No-code neural network Generation and Operation framework)

Other

68 stars 20 forks source link

Hns hanul feature resnet classifier #103

Closed soma0sd closed 11 months ago

soma0sd commented 11 months ago

PR 타입

[x] 기능 추가
[ ] 기능 삭제
[ ] 버그 수정
[x] 의존성, 환경 변수, 빌드 관련 코드 업데이트
[ ] 기타

PR전 체크 사항

[x] 자신의 브랜치로 main 브랜치 최신 내용 반영후 동작 테스트 완료
[x] 다른 작업자의 폴더내 파일 수정하지 않음.
(해당없음) docker-compose.yml 수정한 경우, 수정에 따른 영향 평가

관련 이슈

크기가 큰 모델의 사전학습 가중치는 Model_Mirror에 업로드하여 Docker 컨테이너 빌드 시 컨테이너 내부에 포함할 것

PR 반영 브랜치

HNS_hanul_feature-resnet-classifier -> main

변경 사항

기존 훈련을 통해 얻은 best.pt 대신 github에서 호스팅하는 resnet의 사전학습모델을 image deploy에 포함

테스트 결과

tango_test_image

classification 타입의 프로젝트를 생성
BMS -> Auto NN -> Image Deploy 순으로 수행
내려받은 nn_model.zip 파일 속 resnet.pt의 원본 크기가 약 666MB이 나오는지 확인