[Model Compression] ๋ชจ๋ธ ์์ํ(Model Optimization) with Tensorflow
ยท
๐ฌ ML & Data/โ Q & etc.
๋ฅ๋ฌ๋ ๋ชจ๋ธ์ ๊ฒฝ๋ํํ๋ ๊ฒ์ ๋ชจ๋ธ ํ์ต ์ดํ ์ค์ ๋ฌธ์ ์ ๋ฅ๋ฌ๋ ํด๋ฒ์ ์ ์ฉํ๋ ๊ณผ์ ์ ์์ด์ ์คํ ์๊ฐ, ์์ธก์ ํ์ํ ๋ฆฌ์์ค ์๋ชจ๋์ ์ค์ด๊ธฐ ์ํด์ ํ์ํ ๊ณผ์ ์ด๋ค.๋ชจ๋ธ ๊ฒฝ๋ํ์๋ (๋ด๊ฐ ์๊ณ ์๊ธฐ๋ก๋) ์ธ ๊ฐ์ง ๋ฐฉ๋ฒ์ด ์๋๋ฐ,๋ชจ๋ธ ์์ํ(๋นํธ ์๋ฅผ ์ค์ด๋ ๋ฐฉ์)๋ชจ๋ธ pruning(์ค์ํ์ง ์์ ๋ถ๋ถ์ ๋ฒ๋ฆฌ๋ ๋ฐฉ์)๊ทธ๋ฅ ๋ชจ๋ธ ์ค๊ณ๋ฅผ ์ํ๊ธฐ์ค์ ์ด๋ฏธ ํ์ตํ ๋ชจ๋ธ์ ์์ด์ ๊ฐ์ฅ ์ฌ์ด ์์ํ๋ฅผ ์ฐ์ ํ๊ธฐ๋ก ๊ฒฐ์ ํ์๋ค. Tensorflow ๊ธฐ๋ฐ์ ๋ชจ๋ธ์ ์์ํํ๋ ์์ ๋ฅผ ๊ธฐ๋กํด๋๋ค.Tensorflow๋ก ๊ตฌ์ฑ๋์ด ํ์ตํ๊ณ ๊ฐ์ค์น๋ฅผ .h5 ํ์ฅ์๋ก ์ ์ฅํ ๋ชจ๋ธ1. ๋ชจ๋ธ ์์ํa. ๋ชจ๋ธ ๋ถ๋ฌ์ค๊ธฐimport tensorflow as tfmodel = your_model(parameter)model.load_wei..