SKT-AI/KoGPT2

ONNX Export and Inference

Closed this issue ยท 1 comments

๐Ÿš€ Feature

ONNX export ๋ฐ inference๋ฅผ ์ง€์›ํ•ฉ๋‹ˆ๋‹ค.

Motivation

์ตœ๊ทผ pytorch์—์„œ onnx export์‹œ dynamic_axes๋ฅผ ์ง€์›ํ•ฉ๋‹ˆ๋‹ค. ์‰ฝ๊ฒŒ dynamic sequnce์— ๋Œ€ํ•˜์—ฌ ์ปจ๋ฒ„ํŒ…์ด ๊ฐ€๋Šฅํ•  ๊ฒƒ์œผ๋กœ ๊ธฐ๋Œ€๋ฉ๋‹ˆ๋‹ค.

Pitch

์ž‘์—…ํ•˜์—ฌ PR ํ•˜๋„๋ก ํ•˜๊ฒ ์Šต๋‹ˆ๋‹ค.

์˜ค.. ๊ฐ์‚ฌํ•ฉ๋‹ˆ๋‹ค. ๋จธ์ง€ ํ–ˆ๋„ค์š”..