[ML] 'int' object has no attribute 'to'

train_loader = DataLoader(train_dataset, batch_size=config.batch_size, shuffle=True)

DataLoader를 만들지 않고 데이터를 넣어서 int형으로 나오게됨

to는 list와 같은 배열에 쓸 수 있기 때문에 에러 발생

 

코드를 잘 확인하자.