-
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
미해결
runtime error : Found dtype Long but expected Float
20.11.05 21:34 작성 조회수 1.13k
0
안녕하세요~ 실습 DCGAN 부분에서 D 네트워크를 업데이트 하는 부분에서 에러가 나는데, 라인별로 타입을 확인하면서 수정하려 했는데, 왜 에러가 나는지 모르겠네요 ㅠㅠ 도움 부탁 드립니다.
답변을 작성해보세요.
2
지비
2021.02.19
output, label 둘 다 추가해주니 실행되네요.
---------------------------------------------------------------------------
output = netD(real_cpu).view(-1)
(추가) output = output.type(torch.FloatTensor).cuda()
(추가) label = label.type(torch.FloatTensor).cuda()
errD_real = criterion(output, label)
---------------------------------------------------------------------------
cuda 안 쓰시면 뒤에 .cuda()는 빼셔야 하구요.
1
킴허클베리
질문자2020.11.05
아~! 답을 찾았습니다. 아래 처럼 바꾸니 되네요 ^^
output = netD(real_cpu).view(-1)
#추가하는 줄
output = output.type(torch.FloatTensor)
0
SEULGI로운 취미생활
2021.02.03
real_cpu = data[0].to(device)
b_size = real_cpu.size(0)
label = torch.full((b_size,), real_label, device=device)
output = netD(real_cpu).view(-1)
output = output.type(torch.FloatTensor)
추가해도 저는 똑같네요
답변 3