인프런 커뮤니티 질문&답변

강사님께서 코드를 보면서 '전체 데이터를 가지고 예측을 했기 때문에 굳이 미니배치 기반으로 예측을 안해도 된다' 고 언급하신 부분이 이해가 잘 가지 않습니다.

(9분 ~ 9분 20초 경)

저희가 훈련 데이터로 모델을 학습할 때

미니배치 기준으로 학습을 했으니까 예측을 할 때에도 미니배치 기준으로 해야하지 않나요?

강의 내에서 아래 코드를 굳이 안해도 된다고 언급

09_SIMPLE_ACTUAL_EXAMPLE_MULTI-LABEL_CLASSIFICATION.ipynb

y_pred_list = list()

x_test_batch_list = X_test_tensor.split(minibatch_size, 0)

model.eval()

with torch.no_grad():

    for x_minibatch in x_test_batch_list:

        y_test_pred = model(x_minibatch)

        y_test_pred = torch.argmax(y_test_pred, dim=1)

        y_pred_list.extend(y_test_pred.detach().tolist())

y_pred_list = torch.tensor(y_pred_list)