- Today
- Total
hye-log
[๋ถ์คํธ์บ ํ AI Tech]WEEK 07_DAY 29 ๋ณธ๋ฌธ
๐ฅช ๊ฐ๋ณํ์ต
[8] Training & Inference 2 - Process
1. Training Process
1) Training ์ค๋น
2) Training ํ๋ก์ธ์ค์ ์ดํด
- train : train mode์์ Dropout, BatchNorm ๋ฑ์ ์ํฅ์ ๋ผ์น ์ ์๊ธฐ ๋๋ฌธ์ mode ์ค์ ์ด ํ์
train(mode=True)
- optimizer.zero_grad() : optimizer๊ฐ backward ๋ ๋ model ์์ ์๋ parameter๋ฅผ ์ ๋ฐ์ดํธํ๋๋ฐ, batch iteration์ด ๋์๊ฐ๋ฉด์ zero_grad๋ฅผ ํตํด์ ์ด์ batch์ grad๊ฐ ๋จ์์์ ์ ์๊ธฐ ๋๋ฌธ์ ์ด๊ธฐํ๊ฐ ํ์ํจ
- loss : criterion๋ผ๋ loss ํจ์๋ฅผ ํตํด์ outputs, labels๋ฅผ ๋ฃ๊ณ loss๋ฅผ ๊ตฌํจ
loss = criterion(outputs, labels)
loss.backward()
- optimizer.step() : optimizer๊ฐ input์ผ๋ก ๋ฐ์ parameter์ gradient๊ฐ update ๋์ด ์์ผ๋ฏ๋ก ์ค์ ๋ฐ์ดํฐ์ ๋ฐ์ํจ
optimizer.step()
2. Inference Process
1) Inference ํ๋ก์ธ์ค์ ์ดํด
- model.eval() : evaluation mode๋ก train์ False ํ ๊ฒ๊ณผ ๊ฐ์
eval()
- with torch.no_grad() : with ๋ฌธ ์์ ์๋ gradient์ enable๋ฅผ False๋ก ๋ง๋ค์ด ์ค
๐ฅช ์ค๋์ ํ๊ณ
7์ฃผ์ฐจ์ ์ฒซ ๋ ์ด์ 10์์ ๋ง์ง๋ง ๋ ! ๋ฐ์ผ๋ฆฌ ์คํฌ๋ผ ๋ ๋์จ ์๋ก์ด ๋ชจ๋ธ์ ์๋ํด๋ณด๋ ค๊ณ ์ค์นํ๋ค๊ฐ ๋ญ๊ฐ.. ์ ์ ์๋ ์๋ฌ๊ฐ ๋ฐ์ํด์ ์๋ฒ๋ฅผ ์ด๊ธฐํ์์ผฐ๋ค... ๊ทธ๋ฌ๋ ๋ฐ๋์ ์ฝ๋ ๋ค์ ์ ๋ฆฌํ๊ณ ํ์ผ๋ค ๋ค์ด๋ก๋ ๋ฐ๊ณ ์๋ฒ ๋ค์ ์ด๊ณ .. ์ค์ ๋ถํฐ ์ ์ ์๋ ํ๋ฃจ์๋ค..+_+ schedular ๋ฃ์ด๋ณด๋ ค๊ณ ์ด๊ฒ์ ๊ฒ ์ฐพ์๋ณด๋ค๊ฐ ์ ์ ๋์ด์ ํผ์ด์ธ์ ๋ ํด๊ฒฐํ๋ค!! ์๋ฌด๋๋ ํ์ต์ ์๊ฐ์ด ์ค๋ ๊ฑธ๋ฆฌ๋ค๋ณด๋๊น ์ฌ๋ฌ ๊ฐ์ง ์คํ์ ์ ์ฝ์ด ์๊ธฐ๋ ๊ฒ ์์ฝ๊ธฐ๋ ํ๋ฐ(ใ ใ ) ๊ทธ๋๋ ๋ํ ๋๋๊ธฐ ์ ๊น์ง ์ต์ ์ ๋คํ๋๊ฑธ๋ก!!!!!!!(ํ์ดํ )
'Boostcourse > AI Tech 4๊ธฐ' ์นดํ ๊ณ ๋ฆฌ์ ๋ค๋ฅธ ๊ธ
[๋ถ์คํธ์บ ํ AI Tech]WEEK 07_DAY 31 (0) | 2022.11.02 |
---|---|
[๋ถ์คํธ์บ ํ AI Tech]WEEK 07_DAY 30 (0) | 2022.11.01 |
[๋ถ์คํธ์บ ํ AI Tech]WEEK 06_DAY 28 (0) | 2022.10.28 |
[๋ถ์คํธ์บ ํ AI Tech]WEEK 06_DAY 27 (0) | 2022.10.27 |
[๋ถ์คํธ์บ ํ AI Tech]WEEK 06_DAY 26 (0) | 2022.10.27 |