https://github.com/drawcodeboy/convnet-for-mnist
https://github.com/drawcodeboy/convnet-for-mnist
Last synced: about 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/drawcodeboy/convnet-for-mnist
- Owner: drawcodeboy
- Created: 2024-06-06T06:34:45.000Z (almost 2 years ago)
- Default Branch: main
- Last Pushed: 2024-08-20T09:20:04.000Z (over 1 year ago)
- Last Synced: 2025-01-05T23:14:16.537Z (about 1 year ago)
- Language: Python
- Homepage:
- Size: 12.2 MB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# ConvNet-for-MNIST
```
All code has been written by Doby.
© 2024 Doby. All rights reserved.
```
## Description
* MNIST Classification using Convolutional Network
* Motivation: --
* Project Duration: 24/06/06 15:35~19:20 (3H 45M)
* Author: Doby
## Architecture
### 1. Architecture
### 2. Weight Initialization
```
No Initialization (Random, torch.empty)
Accuracy: 97.19%, Test Loss: 0.0929
Weight Initialization (Gaussian Distribution)
Accuracy: 95.17%, Test Loss: 0.1534
Weight Initialization (Xavier)
Accuracy: 97.66%, Test Loss: 0.0770
```
## Usage
### Download Repository
```
# 프로젝트 Clone
git clone https://github.com/drawcodeboy/ConvNet-for-MNIST.git
# 프로젝트 위치로 이동
cd Path1/Path2/ConvNet-for-MNIST
```
### Virtual Environment
```
# 가상환경 만들기
python -m venv .venv
# 가상환경 활성화
.venv\Scripts\activate
# 가상환경에서 패키지 설치
pip install -r requirements.txt
```
### Train
```
# main.py를 실행시켜 아래와 같은 세팅으로 학습시킨 후
# trained_1epoch_convnet.pt라는 이름 파일로 모델 저장
python main.py --mode=train --epochs=1 --lr=1e-4 --batch_size=16 --device=cpu --file_name=trained_1epoch_convnet.pt
```
### Test
```
# trained_1epoch_convnet.pt 모델을 불러와서 데이터셋 샘플을
# --index를 통해 설정하여 해당 샘플에 대한 predict 결과 확인
python main.py --mode=inference --device=cpu --file_name=trained_1epoch_convnet.pt --index=0
```
## Example
