티스토리 뷰
Computer/이게 왜 안되지?
[이왜안?] RuntimeError: stack(): functions with out=... arguments don't support automatic differentiation, but one of the arguments requires grad.
벼랑끝과학자 2023. 3. 4. 22:15class ToDataset(Dataset):
def __init__(self, df):
self.df = df
# self.src = df['src']
# self.tgt = df['tgt']
# self.label = df['pKa']
def __len__(self):
return len(self.df)
def __getitem__(self, index):
src = self.df.iloc[index]['src']
src = src_embedder(torch.tensor(src, requires_grad=False))
tgt = self.df.iloc[index]['tgt']
tgt = tgt_embedder(torch.tensor(tgt, requires_grad=False))
label = self.df.iloc[index]['pKa']
return src, tgt, label
이렇게 Dataset으로 변환하는 class를 만들고 제대로 변환되었는지 확인하기 위해 아래와 같이 확인하려는데
for i, (src, tgt, label) in enumerate(train_generator):
# print(src, tgt, label)
break
심상찮은 에러가 뜬다.
뭐라는지 모르겠고 중요한건 맨 마지막 줄이겠지, 읽어봤지만 생소하다. 위에서 코드를 보면 알겠지만, requires grad가 붙어있어서 문제가 되는것 같아서 requires_grad 옵션을 False로 줬음에도 불구하고 같은 에러가 떴다.
해결법은, 아예 grad에 대한 기록 자체를 detach()시킨 데이터를 return해야 불러올 수 있었다.
즉 class ToDataset에서 return할 데이터는 다음과 같이 tensor에서 grad는 detach시켜야한다.
class ToDataset(Dataset):
...
def __getitem__(self, index):
...
return src.deatch(), tgt.detach(), label
'Computer > 이게 왜 안되지?' 카테고리의 다른 글
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
TAG
- variational autoencoder
- 기계학습
- 파이썬
- manim library
- eigenvector
- 이왜안
- Manimlibrary
- manim
- 선형대수
- MatrixAlgebra
- 백준
- marginal likelihood
- 베이즈정리
- kld
- manimtutorial
- kl divergence
- elementry matrix
- MorganCircularfingerprint
- MLE
- 인공지능
- 3b1b
- eigenvalue
- 오일석기계학습
- ai인공지능
- 제한볼츠만머신
- vae
- Matrix algebra
- 최대우도추정
- 3B1B따라잡기
- ai신약개발
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | |||||
3 | 4 | 5 | 6 | 7 | 8 | 9 |
10 | 11 | 12 | 13 | 14 | 15 | 16 |
17 | 18 | 19 | 20 | 21 | 22 | 23 |
24 | 25 | 26 | 27 | 28 | 29 | 30 |
글 보관함