①, ①′ 일반적으로 배열이나 행렬과 같은 리스트(list)를 사용하는 방법은 다음과 같습니다.

② for 구문을 사용하여 레코드(행, 가로줄)를 하나씩 가져옵니다. 이때 for x, y in train:과 같이 in 앞에 변수를 두 개 지정해 주면 레코드에서 요소 두 개를 꺼내 오겠다는 의미입니다.

▲ 그림 5-26 for in 구문

③ 모델이 데이터를 처리하기 위해서는 모델과 데이터가 동일한 장치(CPU 또는 GPU)에 있어야 합니다. 코드 5-7에서 model.to(device)가 GPU를 사용했다면, images.to(device), labels.to(device)도 GPU에서 처리되어야 합니다. 참고로 CPU에서 처리된 데이터를 GPU 모델에 적용하거나 그 반대의 경우 런타임 오류가 발생합니다.