Inflearn brand logo image

인프런 커뮤니티 질문&답변

미야옹님의 프로필 이미지
미야옹

작성한 질문수

유니티 머신러닝 에이전트 완전정복 (기초편)

ML-Agents 환경 학습 1

onnx에 대해서

작성

·

28

0

MLAgent환경학습에 보면 학습이 끝나면 결과로

onnx파일이 생성된다고 나와있는데

DQN을 직접짜는 실습2에서도 맨마지막에 학습이 종료되면 onnx파일이 생성되는건가요? 제 경우엔 생성되지 않던데..

1. 직접짜는 방식으로도 onnx는 생성되는게 맞는지

2. 생성되지 않는다면 직접짜는 방식으로 어떻게 생성할수 있는지

이 두가지가 궁금합니다.

답변 1

0

민규식님의 프로필 이미지
민규식
지식공유자

안녕하세요!

코드를 직접 짠 경우에는 ckpt나 pth 같이 pytorch 모델 저장 형식으로 저장됩니다!

onnx로 사용하고 싶다면 이 파일을 직접 onnx로 변환해야하는데 제가 몇년 전에 시도했을 때는 onnx 변환까지는 됐는데 Unity 프로젝트에 적용했을 때 에러가 나는 문제가 있었습니다.

아마 단순 변환이 아닌 듯 하여 해당 부분에 대해서는 공식 코드의 onnx 생성 코드를 참고해야 할 것 같습니다!

미야옹님의 프로필 이미지
미야옹

작성한 질문수

질문하기