데이터 로더1 16. 데이터 로더 1. 데이터 로더(DataLoader)데이터 양이 많을 때 배치 단위로 학습하는 방법 2. 손글씨 인식 모델 만들기import torchimport torch.nn as nnimport torch.optim as optimimport matplotlib.pyplot as pltfrom sklearn.datasets import load_digitsfrom sklearn.model_selection import train_test_split# 코랩에서 GPU 환경으로 변경device = 'cuda' if torch.cuda.is_available() else 'cpu'print(device)digits = load_digits()x_data = digits['data']y_data = digits['tar.. 2024. 6. 13. 이전 1 다음