Fix mnist

[yangjin119]

작업 개요

• 작업 내용: mnist/main.py 소스 코드 수정 및 하이퍼파라미터 조정
• 사용 도구: nano 편집기를 사용하여 로컬에서 수정 진행

변경 사항

1. 하이퍼파라미터 수정

기존 학습 횟수(epochs)를 변경하였습니다.

• 기존: default=10
• 변경: default=14

2. 모델 구조 변경 (Net 클래스)

신경망 구조에 활성화 함수(F.relu)를 추가하여 레이어를 보강했습니다.

# 수정된 코드 부분
x = self.conv2(x)
x = F.relu(x)  # 추가된 라인
x = F.max_pool2d(x, 2)