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
- 가중치 없이 모델 로드
- mqtt
- 모델 freeze
- Foundation Transformers
- mPLUG-2
- cnn
- 주식
- KOSMOS-2
- layer 추출
- Instruction dataset
- 모델 동결
- Multimodal Large Language Model
- 논문 작성
- Video Understanding
- 특정 layer 추출
- vsocde 익스텐션
- 파라미터 수 확인
- MLLM
- 모델 구조 변경
- 파라미터 수
- 주식 용어정리
- 가상환경
- def train
- 논문 작성 요령
- pretrained model layer
- DeepNet
- 강화학습
- def validation
- 논문리뷰
- 특정 layer 동결
Archives
- Today
- Total
시작은 미약하였으나 , 그 끝은 창대하리라
[Pytorch 스크래치 코드] Evaluation 코드 본문
❏ Evaluation code
✓ train, validation 을 통한 early stopping을 수행하였으면 나온 모델을 통해 실제 예측을 수행해야하는 마지막 단계에 도달한다.
이를 위해 아래와 같이 코드를 적용하면 된다.
### Evaluation code ###
### ###
# 예측값 저장하는 빈 리스트
all_predictions = []
# 모델을 평가모드로 설정.
model.eval()
for inputs in test_dataloader:
with torch.no_grad():
inputs=inputs[0].float()
x = inputs.to(device)
outputs = model(x)
max_values, max_indices = torch.max(outputs, 1) # 각 행마다 가장 큰 값을 찾아서 반환함.
# 예측 결과 저장
all_predictions.extend(max_indices.cpu().numpy())
print('저장된 파일 길이 확인: ',len(all_predictions))
✓ 기타 : DataLoader에서 batch_size가 1이나 32, 64 여도 " torch.max( ,dim=1) " 을 수행함으로 잘 작동함.
✓ inputs에 [0]을 취해주지 않으면 리스트로 나오기때문에 model에 못들어감.
✓ Train, Validation 코드 참고.
https://put-idea.tistory.com/90
https://put-idea.tistory.com/91
'인공지능 > 딥러닝 스크래치 코드' 카테고리의 다른 글
[Pytorch 스크래치 코드] 회귀문제 Train, Validation 함수 (1) | 2023.12.17 |
---|---|
[Pytorch 스크래치 코드] 분류문제 Train, Validation 함수 (0) | 2023.12.16 |
[Pytorch 스크래치 코드] Train Test split (1) | 2023.12.10 |
[Pytorch 스크래치 코드] 실험 재현을 위한 Seed 고정 (0) | 2023.12.09 |
[Pytorch 스크래치 코드] Early Stopping (0) | 2023.12.09 |
Comments