티스토리 뷰

기타

[ONNX] aten::unflatten 관련 에러

developer0hye 2023. 8. 21. 22:08

트랜스포머의 Decoder Block을 포함하는 모델을 ONNX로 변환하려했다.

 

이미 작년에 개발한 모델이였고, ONNX 변환도 잘됐던 모델이다.

 

그런데 갑자기 안된다...!

 

Torch 2.x 버전 지우고 1.x 버전으로 설치하니 됐다.

 

크흠...

댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2025/01   »
1 2 3 4
5 6 7 8 9 10 11
12 13 14 15 16 17 18
19 20 21 22 23 24 25
26 27 28 29 30 31
글 보관함