728x90
반응형
에러 발생 환경
pytorch model -> onnx로 변환시에 발생
문제 원인
output 이 list, dictionary 등 tensor 가 아닌 다른 형태이기 때문에 발생
해결 방안
model output을 tensor로 바꿈
728x90
반응형
에러 발생 환경
pytorch model -> onnx로 변환시에 발생
문제 원인
output 이 list, dictionary 등 tensor 가 아닌 다른 형태이기 때문에 발생
해결 방안
model output을 tensor로 바꿈