Notice
Recent Posts
Recent Comments
Link
| ์ผ | ์ | ํ | ์ | ๋ชฉ | ๊ธ | ํ |
|---|---|---|---|---|---|---|
| 1 | 2 | |||||
| 3 | 4 | 5 | 6 | 7 | 8 | 9 |
| 10 | 11 | 12 | 13 | 14 | 15 | 16 |
| 17 | 18 | 19 | 20 | 21 | 22 | 23 |
| 24 | 25 | 26 | 27 | 28 | 29 | 30 |
| 31 |
Tags
- feature map
- BFS
- numpy
- ์ ๊ทํ
- depthwise convolution
- ์ธ์ ๋ฆฌ์คํธ
- ๊ทธ๋ํ
- Inductive Bias
- ๋ฐ์ดํฐ๋ชจ๋ธ๋ง
- dfs
- ๋ฐฑ์ค
- bottleneck
- SQLD
- 1x1 Convolution
- Two Pointer
- Depthwise Separable Convolution
- pytorch
- get_dummies()
- SQL
- ์๋ณ์
- CROSS JOIN
- dp
- outer join
- skip connection
- mobilenet
- resnet
- ์ธ์ ํ๋ ฌ
- ์ฐ์ฐ๋ ๊ฐ์
- SQLD ํ๊ธฐ
- ์ํฐํฐ
Archives
- Today
- Total
๋ชฉ๋ก2024/03/28 (1)
SJ_Koding
๊ฐ์ธ ๊ธฐ๋ก์ฉ PyTorch ์ค๋ฅ ๋ชจ์
1. timm์์ model๋ฅผ loadํ ๋ค, ์ ์ฅ๋ ptํ์ผ์ ๋ถ๋ฌ์ inference๋ฅผ ์ํค๋ฉด ์ฑ๋ฅ์ด ํฌ๊ฒ ๊ฐ์ํ์. --> model.eval()๋ฅผ ๋ฐ๋์ ์คํํด์ค์ผํจ. with torch.no_grad๋ parameter update๋ฅผ ํต์ ํ ๋ฟ, forward ๊ณผ์ ๊น์ง์ update๋ฅผ ํต์ ํ์ง๋ ์๋๋ค. timm์ default mode๋ก train mode๋ก ์ค์ ๋์ด์์ผ๋ฏ๋ก eval()์ ํตํด ๋ชจ๋๋ฅผ ๋ฐ๊ฟ์ค์ผํ๋ค. eval()๋ชจ๋๋ BatchNormalization์ ํ๋ผ๋ฉํฐ๋ฅผ Train์ ์ ํ ํ ๊ฐ์ ๊ทธ๋๋ก ๊ฐ์ ธ์ค์ง๋ง, train()์ ์ ๋ ฅ ๋ฐ์ดํฐ์ ๋ฐ๋ผ BN์ parameter๋ฅผ ๋ณํ์ํค๊ธฐ ๋๋ฌธ์, ์ฑ๋ฅ์ด ์๋ง์ด ๋์๋ ๊ฒ์ด๋ค. (์ถ๊ฐ๋ก dropout ๋ ์ด์ด๋ ์์ ํ ๋ฌด์ํด์ค๋ค.) ์๋ง ..
PyTorch Code/Pytorch
2024. 3. 28. 14:08