• 카테고리

    질문 & 답변
  • 세부 분야

    딥러닝 · 머신러닝

  • 해결 여부

    해결됨

data의 gpu처리 질문

23.12.16 17:57 작성 조회수 115

0

  1. cnn에서는 다음과 같이 dataloader로 부터 얻은 data를 gpu로 변경 했습니다.

    seq, target = data[0].to(device),data[1].to(device),

    하지만 RNN에서는 다음과 같이 cpu로 처리 합니다. 이유가 무엇 인가요??

     

    seq, target = data

 

답변 1

답변을 작성해보세요.

0

안녕하세요!

특별한 이유는 없습니다.

GPU 사용하셔도 되요~

 

감사합니다.