pytorch 오류

pytorch에서 nn.Module을 상속한 클래스를 만들 때, init을 해주지 않아서 발생하는 오류 class aa(nn.Module): def __init__(self): super(aa, self).__init__() ## 이거 추가 self.a = 1 self.b = 2 super()추가하여 수정

torch.expand 함수를 사용할 때, 차원이 1이 아닌 축에 expand를 적용해서 발생한 오류 오류 ex) 에서 1이 아닌 dim을 확장하려고 하면 오류 발생 차원이 1일 때만 정상 동작 이를 대체하여 torch.repeat과 torch.repeat_interleave를 사용할 수 있음. 둘 다 텐서를 반복하여 축을 확장하는 함수이지만, repeat는 행, 열의 값을 무조건 1회 반복해야 함. 더 자세한 예시는 아래 링크에 ) https://cccaaa.tistory.com/33 torch tensor 반복 확장하기 expand vs. repeat vs. repeat_interleave 차이 pytorch에서 텐서를 반복하여 확장하는 방법들을 정리하겠음. 1) torch.expand() dim=1..

custom loss 만들어서 쓰는데, TypeError: 'Tensor' object is not callable 발생함 원인 : loss function이름과 return하는 인자 값 이름이 같아서 발생 loss = loss(pred, ground_truth ) 하면 안됨 이름이 똑같으면 안되서 loss = custom_loss(pred, ground_truth ) 해야 됨~~ ref : https://discuss.pytorch.org/t/typeerror-tensor-object-is-not-callable/102931 TypeError: ‘Tensor’ object is not callable Hello,everyone. Newbie here, trying to learn pytorch. Rec..

파이토치 버전문제임 본인 gpu에 맞는 cuda 설치했으면 파이토치 재설치 또는 본인 cuda와 맞는 파이토치 다운그레이드