배움 - 차근차근 기록하자/[개발] 딥러닝 🍩
libtorch serialize (RT Convert) 오류 1
inference library framework 변경 중 발생한 오류.m_Engine_RTs 객체 초기화가 안되어 발생하였다. 해결 방법 * setDimensions 파라미터 순서 변경-> 주석달린 TensorFlow의 경우에는 NHWC 순서, 주석이 없는 PyTorch의 경우에는 NCHW 순서 (N: 배치, H: 높이, W: 너비, C: 채널 순서) * inputLayer, outputLayer 이름 변경-> 아래 사진의 좌측은 tensorflow, 우측은 pytorch 이다. 같은 모델이더라도 layer 이름이 바뀌어서 맞춰줘야 한다. RT convert 완료
2024. 9. 20. 17:04