기본적인 실수였다...
모델에 넣을 때 Variable로 항상 감싸주자.
from torch.autograd import Variable
input = Variable(x)
model(input)
'데이터분석 with 파이썬 > Pytorch' 카테고리의 다른 글
| python 상대경로 import 실행하기 (0) | 2018.02.13 |
|---|---|
| DataLoader (0) | 2018.02.02 |
데이터분석 with 파이썬/Pytorch
기본적인 실수였다...
모델에 넣을 때 Variable로 항상 감싸주자.
from torch.autograd import Variable
input = Variable(x)
model(input)
| python 상대경로 import 실행하기 (0) | 2018.02.13 |
|---|---|
| DataLoader (0) | 2018.02.02 |