dependency
현재 설치된 cuda 확인
pytorch==2.0.1 CUDA==11.7
설치 완료
LnNet5
Structure
| Layer | Feature Map | Size | Kernel Size | Stride | Activation | |
|---|---|---|---|---|---|---|
| Input | Image | 1 | 32x32 | - | - | - |
| 1 | Conv | 6 | 28x28 | 5x5 | 1 | tanh |
| 2 | Avg Pooling | 6 | 14x14 | 2x2 | 2 | tanh |
| 3 | Conv | 16 | 10x10 | 5x5 | 1 | tanh |
| 4 | Avg Pooling | 16 | 5x5 | 2x2 | 2 | tanh |
| 5 | Conv | 120 | 1x1 | 5x5 | 1 | tanh |
| 6 | FC | - | 84 | - | - | tanh |
| Output | FC | - | 10 | - | - | softmax |
Train
python main.py --mode train --output_dir ./output
Test
python main.py --mode test --output_dir ./output
--checkpoint ./output/MNIST/model_epoch3.pt
