Skip to content

Commit

Permalink
[Add] 3주차 학습코드 흐름도
Browse files Browse the repository at this point in the history
  • Loading branch information
BowonKwon committed Sep 25, 2023
1 parent 3c6b8b8 commit b5125ea
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions 주중수업/3주차/1_train_follow.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
# 패키지 임포트

# hyperparameter 선언

# 모델 설계도 그리기

# 설계도 바탕으로 모델 만들기 <- hyperparameter 사용

# 데이터 불러오기
# dataset 설정(학습, 테스트 데이터)
# dataloader 설정(학습, 테스터 데이터)

# loss 선언
# optimizer 선언

# 학습을 위한 반복 (loop) for / while
# 입력할 데이터를 위해 데이터 준비 (dataloader)
# 데이터와 타깃을 디바이스에 올리기
# 모델에 데이터 넣기
# 모델의 출력과 정답을 비교하기 (loss 사용)
# 역전파 수행
# Loss 바탕으로 가중치 업데이트 진행 (optimizer)
# 그래디언트 초기화

# 평가(로깅, 출력), 저장

0 comments on commit b5125ea

Please sign in to comment.