pytorch
-
RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn프로그래밍/에러 처리 2022. 12. 6. 17:26
Pytorch Lightning에서 model의 모든 layer를 freeze하는 테스트를 진행했다. 이때 freeze 사용된 코드는 아래와 같다. for params in model.parameters(): params.requires_grad = False 그러나 위의 코드를 실행했을 경우 Trainer.fit 부분에서 아래의 에러가 발생했다. RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn 이리저리 살펴본 결과 결국 pytorch lightning에서 모든 레이어가 params.requires_grad = False일 경우 evaluate 모드와 다름이 없어 에러가 발생하는 것으로 보인다. 이후 ..
-
Pytorch Lightning 사용가이드 및 코드 예시(Scene Classification Dataset)Data Science/Pytorch 2022. 11. 17. 02:24
1. Pytorch Lightning이란? Tensorflow의 Keras와 같이 Pytorch를 위한 라이브러리이다. pytorch lightning의 전반적 동작 구조는 아래와 같다. 2. 라이브러리 준비 from glob import glob import torch import torch.utils.data as data from torchvision import transforms import PIL from torchvision.datasets import ImageFolder from torchvision.models import resnet18 from torch.optim import Adam from torch.nn import functional, CrossEntropyLoss from..
-
Pytorch Resnet 예시코드(CNN, Resnet18, Cats and Dogs Dataset, Augmentation)Data Science/Pytorch 2022. 11. 10. 00:44
1. Data Set 준비하기 1.1 Data .zip 파일 다운받기 !wget --no-check-certificate \ https://storage.googleapis.com/mledu-datasets/cats_and_dogs_filtered.zip \ -O ./cats_and_dogs_filtered.zip 1.2 .zip 파일 압축 풀기 ! unzip -q cats_and_dogs_filtered.zip -d ./ 1.3 파일 구조 "cats and dogs filtered" 폴더 내부에서 "train", "validation" 폴더로 나누어진다. 각각 "cats", "dogs" 폴더를 가지며, 그 내부에는 해당하는 이미지들이 들어있다. 2. Data Preprocessing 2.1 이미지 경로..