작성
·
28
0
MLAgent환경학습에 보면 학습이 끝나면 결과로
onnx파일이 생성된다고 나와있는데
DQN을 직접짜는 실습2에서도 맨마지막에 학습이 종료되면 onnx파일이 생성되는건가요? 제 경우엔 생성되지 않던데..
1. 직접짜는 방식으로도 onnx는 생성되는게 맞는지
2. 생성되지 않는다면 직접짜는 방식으로 어떻게 생성할수 있는지
이 두가지가 궁금합니다.
답변 1
0
안녕하세요!
코드를 직접 짠 경우에는 ckpt나 pth 같이 pytorch 모델 저장 형식으로 저장됩니다!
onnx로 사용하고 싶다면 이 파일을 직접 onnx로 변환해야하는데 제가 몇년 전에 시도했을 때는 onnx 변환까지는 됐는데 Unity 프로젝트에 적용했을 때 에러가 나는 문제가 있었습니다.
아마 단순 변환이 아닌 듯 하여 해당 부분에 대해서는 공식 코드의 onnx 생성 코드를 참고해야 할 것 같습니다!