728x90 pytorch2 torch.Tensor.item https://pytorch.org/docs/stable/generated/torch.Tensor.item.html torch.Tensor.item — PyTorch 2.1 documentation Shortcuts pytorch.org 모델에서 예측한 값(y_pred)과 참 값(y_trgt)값이 같은 것의 개수를 구하려 할 때 (y_pred==y_trgt).sum().item()을 쓰는 코드를 발견해서 디버깅을 해보면서 torch.Tensor.item()이 텐서 값을 숫자만으로 나타낼 수 있다는 것을 알았다. 이는 dictionary에서 (key, value) 값으로 나타내는 items과 다른 의미를 나타낼 수 있다는 것을 알았다. python items code car = { "brand": "BMW.. 2023. 11. 9. [PyTorch] Dataset & Dataloaders 모델에 데이터를 넣을려면 어떻게 해야할까? 자료를 모은다(collecting) → 자료를 정리한다(cleaning) → 전처리 작업 (pre processing) → 데이터(Data) 완성! 그럼 이것을 바로 모델에 넣으면 될까? NO! Dataset 클래스(class)를 통해 목적에 맞는 데이터 값을 받고 이것을 DataLoader라는 것을 통해 Model에 맞는 batch 등 Model input에 맞게 다듬은 다음에 Model에 넣는 것이다. 뭔 말이지...? 그래서 우선 하나씩 알아보자 Dataset 클래스 - 데이터 입력 형태를 정의하는 클래스 - 데이터를 입력하는 방식을 표준화 한다는 것에 의의가 있다. - Image, Text, Audio 등에 따른 다른 입력 정의를 내린다. - 클래스 안에.. 2022. 12. 24. 이전 1 다음 728x90