마지막 데이터 짤짤이 컷하는 방법 in python

vaild dataset 배치로 만든 마지막 텐서 크기가  갯수가 모자라서 8개로 설정된거라 vaild dataset 만드실때 파라미터로 drop_last = True 추가해주시면 됩니다. 즉 데이터가 16개씩 들어가는데, 마지막에 16개가 들어가야하나 8개만 들어가서 생기는 오류.

 

val_dataloader = torch.utils.data.DataLoader(val_dataset, batch_size=16, shuffle=True, num_workers=2)
여기에 drop_last = True 를 인자로 넣어보자.