[Huggingface] Model Memory Calculator, GPU ์ผ๋ง๋ฉด ๋๋?
ยท
๐ฃ๏ธ Natural Language Processing
Model Memory Calculator, GPU ์ผ๋ง๋ฉด ๋๋? llama3, gemma2, florence ๋ฑ llama1(2023.2.24)์ด ๋์จ ์ง ๋ฒ์จ 1๋
์ด ๋์ด๊ฐ๋๋ฐ ์์ง ์คํ llm์ ์ธ๊ธฐ๋ ์์ ์ค ๋ชจ๋ฅด๊ณ ์๋ค. ์๋ ๋ ์ธ๊ธฐ๊ฐ ๋๊ณ ์๋ค. ํ์ต ํ์ดํ๋ผ์ธ์ ๋์ฑ ์ฝ๊ณ ๊ฒฌ๊ณ ํด์ง๊ณ ๋ชจ๋ธ inference๋ ๋์ฑ ๋ฆฌ์์ค ์๋ ๋ค ๋ฐ๋ฌํ๊ณ ์๋ค. ๊ทธ๋ฐ๊ณ ๋ก ๋์ ๋ฆฌ์์ค์ ๋ง๋ ๋ชจ๋ธ์ ๋ฌด์์ด๊ณ ์ต๋์น๋ก ๋๋ฆด ์ ์๋ ๊ฒ๋ค์ด ๊ถ๊ธํ ๊ฒ์ด๋ค. ๋จผ์ 2b, 7b, 9b์ด ์ซ์์ ๋ํด ๊ฐ๋จํ ์ค๋ช
ํ๋ฉด ๋ชจ๋ธ์ด ํ์ตํ parameter์ ์์ด๋ค. ๊ฐ๋จํ ์ด์ผ๊ธฐํ๋ฉด ๋ชจ๋ธ์ด ํํํ ์์๋ ๊ฒฝ์ฐ์ ์๊ฐ ์ด๋งํผ ๋ง๋ค๋ ๊ฒ์ด๋ค. ๊ณผ๊ฑฐ BERT ๋ชจ๋ธ์ ๋จ์๊ฐ 3M, 5M ๋ฐฑ๋ง ๋จ์๋ผ๋ฉด ์ง๊ธ์ ์์ญ์ต ๋จ์๋ก ๋์ด์..